排序搜索计数
如何通过第n列对一个数组进行排序
Z = np.random.randint(0,10,(3,3))
【知识点:排序】
- (提示: argsort)
#【答案】
Z = np.random.randint(0,10,(3,3))
print (Z)
'''[[1 6 0]
[7 0 3]
[0 6 0]]'''
print (Z[Z[:,2].argsort()])
'''[[1 6 0]
[0 6 0]
[7 0 3]]'''
从arr
中提取所有奇数。
arr = np.arange(10)
【知识点:搜索】
- 如何从一维数组中提取满足指定条件的元素?
#【答案】
import numpy as np
arr = np.arange(10)
# 方法1
index = np.where(arr % 2 == 1)
print(arr[index])
# [1 3 5 7 9]
# 方法2
x = arr[arr % 2 == 1]
print(x)
# [1 3 5 7 9]
将arr
中的偶数元素替换为0。
arr = np.arange(10)
【知识点:搜索】
- 如何用numpy数组中的另一个值替换满足条件的元素项?
#答案
import numpy as np
arr = np.arange(10)
index = np.where(arr % 2 == 0)
arr[index] = 0
print(arr)
# [0 1 0 3 0 5 0 7 0 9]
** 将 arr
中的所有偶数元素替换为0,而不改变arr。**
arr = np.arange(10)
【知识点:搜索】
- 如何在不影响原始数组的情况下替换满足条件的元素项?
#答案
import numpy as np
arr = np.arange(10)
# 方法1
x = np.where(arr % 2 == 0, 0, arr)
print(x)
# [0 1 0 3 0 5 0 7 0 9]
print(arr)
# [0 1 2 3 4 5 6 7 8 9]
# 方法2
x = np.copy(arr)
x[x % 2 == 0] = 0
print(x)
# [0 1 0 3 0 5 0 7 0 9]
print(arr)
# [0 1 2 3 4 5 6 7 8 9]
获取给定数组a中前5个最大值的位置。
a = np.random.uniform(1, 50, 20)
【知识点:搜索】
- 如何从numpy数组中获取最大的n个值的位置?
#答案
import numpy as np
np.random.seed(100)
a = np.random.uniform(1, 50, 20)
print(a)
# [27.62684215 14.64009987 21.80136195 42.39403048 1.23122395 6.95688692
# 33.86670515 41.466785 7.69862289 29.17957314 44.67477576 11.25090398
# 10.08108276 6.31046763 11.76517714 48.95256545 40.77247431 9.42510962
# 40.99501269 14.42961361]
# 方法1
b = np.argsort(a)
print(b)
print(b[-5:])
# [18 7 3 10 15]
# 方法2
b = np.sort(a)
b = np.where(a >= b[-5])
print(b)
# (array([ 3, 7, 10, 15, 18], dtype=int64),)
# 方法3
b = np.argpartition(a, kth=-5)
print(b[-5:])
# [18 7 3 10 15]
删除一维numpy数组中所有NaN值。
a = np.array([1, 2, 3, np.nan, 5, 6, 7, np.nan])
【知识点:逻辑函数、搜索】
- 如何删除numpy数组中的缺失值?
#答案
import numpy as np
a = np.array([1, 2, 3, np.nan, 5, 6, 7, np.nan])
b = np.isnan(a)
c = np.where(np.logical_not(b))
print(a[c])
# [1. 2. 3. 5. 6. 7.]
集合操作
获取数组a和数组b之间的公共项。
a = np.array([1, 2, 3, 2, 3, 4, 3, 4, 5, 6])
b = np.array([7, 2, 10, 2, 7, 4, 9, 4, 9, 8])
【知识点:集合操作】
- 如何获取两个numpy数组之间的公共项?
#答案
import numpy as np
a = np.array([1, 2, 3, 2, 3, 4, 3, 4, 5, 6])
b = np.array([7, 2, 10, 2, 7, 4, 9, 4, 9, 8])
x = np.intersect1d(a, b)
print(x) # [2 4]
从数组a中删除数组b中的所有项。
a = np.array([1, 2, 3, 4, 5])
b = np.array([5, 6, 7, 8, 9])
【知识点:集合操作】
- 如何从一个数组中删除存在于另一个数组中的项?
#答案
import numpy as np
a = np.array([1, 2, 3, 4, 5])
b = np.array([5, 6, 7, 8, 9])
x = np.setdiff1d(a, b)
print(x) # [1 2 3 4]