使用 . 創建兩個陣列numpy。一個叫具有價值觀的學生。
['Janet', 'Adriana', 'Manual', 'Mohamed', 'Leann']
另一個稱為等級作為值：
[[93, 85], [78, 80], [94, 93], [75, 90], [92, 87]]
從學生所在的年級中選擇所有行'Adriana' or 'Mohamed'
我該如何解決這個問題？
uj5u.com熱心網友回復：
你可以使用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]])
轉載請註明出處，本文鏈接：https://www.uj5u.com/houduan/537355.html
上一篇：結構內的動態分配矩陣