如何将str系列转换为二维ndarray？

Question

enter image description here

这是我的数据

我想将其转移到二维ndarray中，我尝试了很多方法，例如np.from_string，pd.pd.to_numeric，但我无法解决

非常感谢。

Answer 1

将list comprehension与split一起使用并转换为np.array：

df = pd.DataFrame({'pixels':['70 80 82 72','151 1050 147 155']})
print (df)
             pixels
0       70 80 82 72
1  151 1050 147 155

arr = np.array([x.split() for x in df['pixels']]).astype(int)
print (arr)
[[  70   80   82   72]
 [ 151 1050  147  155]]

如果文件中有数据：

arr = np.genfromtxt('my_file.csv', delimiter=' ', dtype=np.int64)
print(arr)
[[  70   80   82   72]
 [ 151 1050  147  155]]