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

jjxx2005

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

复制代码

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

请问大神怎么做 谢谢

最佳答案

月排行榜 / 总排行榜

Twilight6

2020-6-20 12:48:50

这样即可：

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

复制代码

跳转到最佳答案楼层

Twilight6

这样即可：

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

复制代码

塔利班

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

复制代码