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

Question

我有一个 numpy 重新数组，其中包含几个整数列和一些字符串列。字符串列中的数据 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])