将一些数据帧信息传输到新矩阵

Question

我正在尝试将数据帧行 0 到 15 和列 col1 到 col 16 的信息“转换”为图像 (16x16)

我正在从 .txt 文件中读取数据框：

df = pd.read_csv('User1/Video1.txt', sep="\s+|\t+|\s+\t+|\t+\s+", header=None, names=headers, engine='python', parse_dates=parse_dates)

                        date arrow  col1  col2  ...  col13  col14  col15  col16
0    2020-11-09 09:53:39.552    ->   0.0   0.0  ...    0.0    0.0    0.0    0.0
1    2020-11-09 09:53:39.552    ->   0.0   2.0  ...    0.0    0.0    0.0    0.0
2    2020-11-09 09:53:39.552    ->   0.0   0.0  ...    0.0    0.0    6.0    6.0
3    2020-11-09 09:53:39.552    ->   0.0   0.0  ...    0.0    0.0    0.0    0.0
4    2020-11-09 09:53:39.586    ->   0.0   9.0  ...    0.0    7.0    0.0    0.0
...                      ...   ...   ...   ...  ...    ...    ...    ...    ...
15   2020-11-09 09:54:06.920    ->   4.0   0.0  ...    4.0    4.0    0.0    0.0

创建空矩阵img = np.zeros((16, 16, 3), dtype=np.uint8) 后，我想遍历数据框以传输列信息。

为此，我想使用df.itertuples，但我在填写括号时遇到了问题。

for row in df.itertuples():
     img[]][] += row[]

你能提供什么建议吗？谢谢。

Answer 1

使用DataFrame.iloc + np.resize

#import numpy as np
np.resize(df.iloc[:16, -16:].to_numpy(), (16, 16, 3))

如果您的列未向右对齐，您可以使用 'col' 过滤

np.resize(df.filter(regex='col').iloc[:16, :].to_numpy(), (16, 16, 3))