在numpy矩阵中令小于0的元素改为0的实例

时间:2021-12-05 19:50:58

如下所示:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
>>> import numpy as np
>>> a = np.random.randint(-5, 5, (5, 5))
>>> a
array([[-4, -4, -5, 2, 1],
  [-1, -2, -1, 3, 3],
  [-1, -2, 3, -5, 3],
  [ 0, -3, -5, 1, -4],
  [ 0, 3, 1, 3, -4]])
# 方式一
>>> np.maximum(a, 0)
array([[0, 0, 0, 2, 1],
  [0, 0, 0, 3, 3],
  [0, 0, 3, 0, 3],
  [0, 0, 0, 1, 0],
  [0, 3, 1, 3, 0]])
# 方式二
>>> (a + abs(a)) / 2
array([[0, 0, 0, 2, 1],
  [0, 0, 0, 3, 3],
  [0, 0, 3, 0, 3],
  [0, 0, 0, 1, 0],
  [0, 3, 1, 3, 0]])
# 方式三
>>> b = a.copy()
>>> b[b < 0] = 0
>>> b
array([[0, 0, 0, 2, 1],
  [0, 0, 0, 3, 3],
  [0, 0, 3, 0, 3],
  [0, 0, 0, 1, 0],
  [0, 3, 1, 3, 0]])
# 方式四
>>> np.where(a > 0, a, 0)
array([[0, 0, 0, 2, 1],
  [0, 0, 0, 3, 3],
  [0, 0, 3, 0, 3],
  [0, 0, 0, 1, 0],
  [0, 3, 1, 3, 0]])

以上这篇在numpy矩阵中令小于0的元素改为0的实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持服务器之家。

原文链接:https://blog.csdn.net/qq_22291287/article/details/79850263