如何B根據陣列索引撰寫串列元素A？附加了所需的輸出。
import numpy as np
A = np.array([[1,2,3],[4,5,6],[7,8,9]])
B = [1,5,6]
所需的輸出是
B=[A[0,0], A[1,1], A[1,2]]
uj5u.com熱心網友回復：
您可以使用numpy.isin查看中的元素是否B存在于A; 然后用于numpy.where查找存在的元素的索引。由于索引由軸分隔，您最終可以解包并zip獲得所需的結果：
out = list(zip(*np.where(np.isin(A, B))))
對于A=np.array([[1,2,3],[4,5,6],[7,8,9]])和B=[1,5,6,7]，輸出：
[(0, 0), (1, 1), (1, 2), (2, 0)]
轉載請註明出處，本文鏈接：https://www.uj5u.com/ruanti/443142.html
上一篇：WebBrowser中的KeyDown(ExcelVBA)
下一篇：為什么用陣列索引陣列會回傳兩行？