当我使用 numpy genfromtxt 函数将字符串转换为整数时，为什么我要转换的内容全为 0

Question

我应该得到不同数量的不同材质的木头。但是我所有的材料都是0。

%%writefile wood.txt

item,material,number

100,oak,33
110,maple,14
120,oak,7
145,birch,3


tree_to_int = dict(oak=1,maple=2,birch=3)

def convert(s):
    return tree_to_int.get(s,0)

data = np.genfromtxt('wood.txt', delimiter=',', dtype=np.int, 
names=True,converters={1:convert})
data

[输出]:

array([(100, 0, 33), (110, 0, 14), (120, 0,  7), (145, 0,  3)],dtype=[('item', '<i4'), ('material', '<i4'), ('number', '<i4')])

Answer 1

原来是因为从"wood.txt"读取的字符串是bytestrings。这就是为什么他们无法在字典中找到的原因。要修复它，只需像下面这样解码字节串

def convert(s):
    return tree_to_int.get(s.decode("utf-8") , 0)

或者，您也可以使用 pandas

import pandas

tree_to_int = pandas.DataFrame([{'material': 'oak', 'material_int': 1}, {'material': 'maple', 'material_int': 2}, {'material': 'birch', 'material_int': 3}])

df = pandas.read_csv('wood.txt')

data = pandas.merge(df, tree_to_int, how='left', on='material')

当我使用 numpy genfromtxt 函数将字符串转换为整数时，为什么我要转换的内容全为 0

when I convert string to integer using numpy genfromtxt function, why is it all 0 for what I want to convert

python

numpy

genfromtxt