python - 如何导入具有混合浮点数和非数字字符串的数组，并将字符串转换为零

Question

我有一个类似于[1.124, 5.152, 6.235, NaN, 5.124, Infinity]. 我想将其导入为[1.124, 5.152, 6.235, 0, 5.124, 0]. 我该怎么做呢？

Answer 1

基于 NaN 和 Inf 的过滤器：

print([float(ele) if ele not in {"NaN","Infinity"} else 0 for ele in arr])

如果您使用的是 numpy.

arr = np.genfromtxt("a.txt", dtype=float,delimiter=",")
print(arr)
arr[~np.isfinite(arr)] = 0
print(arr)
[ 1.124  5.152  6.235    nan  5.124    inf]
[ 1.124  5.152  6.235  0.     5.124  0.