[已解决]Numpy数组值替换问题

jjxx2005

import numpy as np
arr1 = np.arange(1,21).reshape(4,5)
print(arr1[:,3])  # 所有行，第3列
筛选 = arr1[:,3] > 5  # 所有行第3列，大于5的
arr1[筛选>5] = 520
print(arr1)

我想把数组第3列大于5个数替换成520
但是无论怎么做都是把整个数组的值中大于5的替换成520了

请问大神怎么做 谢谢

最佳答案

月排行榜 / 总排行榜

Twilight6

2020-6-20 12:48:50

这样即可：

import numpy as np
arr1 = np.arange(1,21).reshape(4,5)
print(arr1[:,3])  # 所有行，第3列
筛选 = arr1[:,3]>5  # 所有行第3列，大于5的
arr1[:,3][筛选] = 520
print(arr1)

跳转到最佳答案楼层

Twilight6

这样即可：

import numpy as np
arr1 = np.arange(1,21).reshape(4,5)
print(arr1[:,3])  # 所有行，第3列
筛选 = arr1[:,3]>5  # 所有行第3列，大于5的
arr1[:,3][筛选] = 520
print(arr1)

塔利班

>>> arr1
array([[ 1,  2,  3,  4,  5],
       [ 6,  7,  8,  9, 10],
       [11, 12, 13, 14, 15],
       [16, 17, 18, 19, 20]])
>>> b=arr1[:,3]
>>> b[b>5]=520
>>> arr1
array([[  1,   2,   3,   4,   5],
       [  6,   7,   8, 520,  10],
       [ 11,  12,  13, 520,  15],
       [ 16,  17,  18, 520,  20]])
>>>