NumPy 数组元素未更新

Question

我有一个如下的 NumPy 数组：

supp = np.array([['A', '5', '0'], ['B', '3', '0'], ['C', '4', '0'], ['D', '1', '0'], ['E', '2', '0']])

现在，我想将 row[2] 更新为 row[1]/6。 我正在使用..

for row in supp: row[2] = row[1].astype(int) / 6

但第 [2] 行似乎不受影响..

>>> supp
array([['A', '5', '0'],
   ['B', '3', '0'],
   ['C', '4', '0'],
   ['D', '1', '0'],
   ['E', '2', '0']],  
  dtype='<U1')

我正在使用 Python 3.5.2 和 NumPy 1.11.1。

感谢任何帮助。提前致谢

Answer 1

问题是np.array 只有一种类型，自动假定为字符串supp.dtype == '|S1'，因为您的输入仅包含长度为1 的字符串。因此，在您的情况下，numpy 会自动将您更新的输入转换为长度为 1、'0's 的字符串。强制它为泛型类型object，然后它将能够同时具有字符串和整数或浮点数或其他任何东西：

supp = np.array([['A', '5', '0'], ['B', '3', '0'], ['C', '4', '0'], ['D', '1', '0'], ['E', '2', '0']])
supp = supp.astype(object)

for row in supp:
    row[2] = int(row[1]) / 6

结果：

[['A' '5' 0.8333333333333334]
 ['B' '3' 0.5]
 ['C' '4' 0.6666666666666666]
 ['D' '1' 0.16666666666666666]
 ['E' '2' 0.3333333333333333]]

或者，您也可以使用dtype '|Sn'，n 的值更大：

supp = np.array([['A', '5', '0'], ['B', '3', '0'], ['C', '4', '0'], ['D', '1', '0'], ['E', '2', '0']])
supp = supp.astype('|S5')

for row in supp:
    row[2] = int(row[1]) / 6

结果：

[['A' '5' '0.833']
 ['B' '3' '0.5']
 ['C' '4' '0.666']
 ['D' '1' '0.166']
 ['E' '2' '0.333']]

在这种情况下，如果这是你想要的，你仍然只有字符串。