Python ndarray 字符串数组转换为 int

在数据处理过程中，我们有时需要将字符串数组转换为整型数组。numpy中的ndarray可以非常快速地完成这个任务。

以下是将字符串数组转换为整型数组的示例代码：

import numpy as np

str_arr = np.array(['1', '2', '3', '4', '5'])

int_arr = str_arr.astype(np.int)

print(int_arr)

输出结果会是一个整型数组：

[1 2 3 4 5]

使用astype方法将字符串数组转换为整型数组，可以极大地提高程序的执行效率。

注意：如果字符串数组中存在非数字字符，将会抛出ValueError异常。因此建议在转换前先进行数据类型的检查和清洗。

ValueError: invalid literal for int() with base 10: 'abc'

总结：

• numpy中的ndarray可以快速地将字符串数组转换为整型数组。
• 使用astype方法可以实现上述目的。
• 在进行转换前需要进行数据类型的检查和清洗，避免出现异常。