使用pandas读取带有numpy数组的csv

Question

我有一个csv3列文件emotion, pixels, Usage组成的35000行如0,70 23 45 178 455,Training.

我pandas.read_csv以前读过这个csv文件pd.read_csv(filename, dtype={'emotion':np.int32, 'pixels':np.int32, 'Usage':str}).

当我尝试上述内容时,它说ValueError: invalid literal for long() with base 10: '70 23 45 178 455'？我如何读取像素列作为numpy数组？

Answer 1

请尝试以下代码 -

df = pd.read_csv(filename, dtype={'emotion':np.int32, 'pixels':str, 'Usage':str})

def makeArray(text):
    return np.fromstring(text,sep=' ')

df['pixels'] = df['pixels'].apply(makeArray)