我將 csv 文件導(dǎo)入到 numpy 數(shù)組中，需要將其轉(zhuǎn)換為結(jié)構(gòu)化數(shù)組，其中僅第一列作為 dtype 字符串，所有其他 47 列作為 float。如何在單個操作中定義其他 47 列的數(shù)據(jù)類型？我是否必須逐列指定 dtype？
1 回答
繁華開滿天機
TA貢獻1816條經(jīng)驗 獲得超4個贊
您可以像結(jié)構(gòu)化數(shù)組一樣讀取源文件。
假設(shè)您輸入的文件包含：
一個字符串字段，命名為Id，
只有四個 浮點字段，分別命名為F1、F2等。
所以它的內(nèi)容是：
ABCD,160.72,180.21,260.13,451.48
EFGH,252.42,132.21,150.11,612.56
IJKL,541.77,455.21,268.76,543.81
要讀取這樣的文件，您可以使用np.loadtxt方法，將dtype作為結(jié)構(gòu)化類型（定義列表）傳遞，可以在列表理解中生成：
nFloats = 4
a = np.loadtxt('Input.csv', delimiter=',',
dtype=[('Id', 'U10')] + [( f'F{i+1}', '<f4' ) for i in range(nFloats)])
請注意，我傳遞了U10作為Id列的類型（10 個字符）。如果需要，請設(shè)置該字段的其他大小。
結(jié)果是：
array([('ABCD', 160.72, 180.21, 260.13, 451.48),
('EFGH', 252.42, 132.21, 150.11, 612.56),
('IJKL', 541.77, 455.21, 268.76, 543.81)],
dtype=[('Id', '<U10'), ('F1', '<f4'), ('F2', '<f4'), ('F3', '<f4'), ('F4', '<f4')])
當(dāng)然，在代碼的目標版本中相應(yīng)地增加nFloats（可能應(yīng)該是47）。
添加回答
舉報
0/150
提交
取消