使用numpy创建两个数组。一个名为students，值为。
numpy
['Janet', 'Adriana', 'Manual', 'Mohamed', 'Leann']
另一种叫做等级作为价值观：
[[93, 85], [78, 80], [94, 93], [75, 90], [92, 87]]
从学生为'Adriana' or 'Mohamed'的年级中选择所有行我该如何解决这个问题？
'Adriana' or 'Mohamed'
zynd9foi1#
您可以使用numpy.isin。
numpy.isin
import numpy as np students = ['Janet', 'Adriana', 'Manual', 'Mohamed', 'Leann'] grades = [[93, 85], [78, 80], [94, 93], [75, 90], [92, 87]] arr_s = np.asarray(students) arr_g = np.asarray(grades) mask = np.isin(arr_s, ['Adriana', 'Mohamed']) res = arr_g[mask] print(res)
输出量：
array([[78, 80], [75, 90]])
1条答案
按热度按时间zynd9foi1#
您可以使用
numpy.isin。输出量：