numpy中常用的函数总结

不定期更新，多积累

where

• 语法格式

numpy.where(condition[, x, y])

位置函数，通过条件筛选符合要求的数据。$x$和$y$表示候选项，非必需，且两者必须同时给出或不给出。若是满足就用$x$来填充，不满足就用$y$来填充

• 案例展示

import numpy as np
a = np.array([[1,2,3,4],[3,4,5,6],[6,7,8,9]])
a

array([[1, 2, 3, 4],
       [3, 4, 5, 6],
       [6, 7, 8, 9]])

# 返回大于5的位置
np.where(a > 5)

(array([1, 2, 2, 2, 2]), array([3, 0, 1, 2, 3]))

# 大于5的填充为1，小于或等于5的填充为10
np.where(a > 5,1,10)

array([[10, 10, 10, 10],
       [10, 10, 10,  1],
       [ 1,  1,  1,  1]])