python - numpy记录数组中列的数据类型转换

Question

我有一个带有多个整数列和一些字符串列的 numpy recarray。字符串列中的数据由 99% 的整数组成，但 numpy 的东西是字符串，因为“NA”在列中。

所以我有两个问题:

• 如何删除 NA 并将它们更改为 0？

• 如何将字符串列转换为整数，以便我可以拥有包含许多整数列的记录数组？

谢谢。

Answer 1

使用where和astype:

>>> x = np.array([123, 456, "789", "NA", "0", 0])
>>> x 
array(['123', '456', '789', 'NA', '0', '0'], dtype='|S8')
>>> np.where(x != 'NA', x, 0).astype(int)
array([123, 456, 789,   0,   0,   0])