python - 制作包含字符串和整数的 ndarray

Question

如何通过读取 csv 文件来制作 ndarray

12,employed,32,happy,1
21,unemployed,31,poor,0
34,rich,45,unhapppy,0

注意:文件可能很大

输出数组:

[[12,"employed",32,"happy",1]
 [21,"unemployed",31,"poor",0]
 [34,"rich",45,"unhapppy",0]]

在使用 np.genfromtxt(filename,delimiter = ",",dtype = None) 从 csv 文件读取时，它生成元组的一维数组并使用 dtype = int 它使字符串成为 Nan

Answer 1

使用read_csv先然后DataFrame.values转换为 numpy 数组:

import pandas as pd 
df = pd.read_csv('file', header=None)
print(df)
    0           1   2         3  4
0  12    employed  32     happy  1
1  21  unemployed  31      poor  0
2  34        rich  45  unhapppy  0

arr = df.values
print(arr)
[[12 'employed' 32 'happy' 1]
 [21 'unemployed' 31 'poor' 0]
 [34 'rich' 45 'unhapppy' 0]]