ValueError：必须通过二维输入。 形状=(430, 430, 3)

Question

I want to save this as 3d array csv file with header as r,g,b but it is showing ValueError: Must pass 2-d input. 形状=（430、430、3）。

four_img_concat 形状为 (430,430,3)。

import pandas as pd
import numpy as np
ans=np.array(four_img_concat)
headers=np.array(['r','g','b'])
df=pd.DataFrame(ans)
df.to_csv("answ.csv",index=False,header=headers)

Answer 1

如错误所示，您需要将数组从 3-d 转换为 2-d。 您可以通过使用reshape function 将像素总数传递到一个轴 ( 430*430 ) 来做到这一点。

np.random.seed(42)
four_img_concat = np.random.randint(0,255,size=(430,430,3))
print(four_img_concat.shape) # (430, 430, 3)

four_img_concat = np.reshape(four_img_concat, (430*430, -1))
print(four_img_concat.shape)  # (184900, 3)

ans = np.array(four_img_concat)
headers = np.array(['r','g','b'])

df = pd.DataFrame(ans)
df.to_csv("answ.csv",index=False,header=headers)

answ.csv文件

r,g,b
102,179,92
14,106,71
188,20,102
121,210,214
74,202,87
...
...