我試圖解決的問題如下。我得到一個表示串列索引的任意維度的矩陣,然后是一個串列。我想取回一個矩陣,其中串列元素交換為索引。我不知道如何以矢量化的方式做到這一點:即如果z = [[0,1], [1,0]]和list = [20,10],我想要[[20,10], [10,20]]回傳。
uj5u.com熱心網友回復:
當它們都為 時np.array,您可以以自然的方式進行索引:
import numpy as np
z = np.array([[0, 1], [1, 0]])
a = np.array([20, 10])
output = a[z]
print(output)
# [[20 10]
# [10 20]]
轉載請註明出處,本文鏈接:https://www.uj5u.com/caozuo/368027.html
下一篇:大資料的MLE方法如何防止溢位
