我是 numpy 的新手,目前正在從事一個建模專案,我必須根據兩個不同的資料源執行一些計算。但是,直到現在我還沒有弄清楚如何將所有單獨的值相互乘以:
我有兩個資料框
一個二維資料框:
df1 = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
一個一維資料框:
df2 = np.array([1, 2, 3, 4, 5])
我想將第一個資料幀 (df1) 中的所有單個值分別與存盤在第二個資料幀中的所有值相乘,以創建形狀為 5x3x3 的資料立方體/新 3D 資料幀:
df3 = np.array([[[1, 2, 3], [4, 5, 6], [7, 8, 9]], [[2, 4, 6], [8, 10, 12], [14, 16, 18]], ..... ])
我嘗試了不同的方法,但每次我都無法獲得類似 df3 的東西。
x = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
y = np.array([1, 2, 3, 4, 5])
z = y
for i in range(len(z)):
z.iloc[i] = x
for i in range(0, 5):
for j in range(0, 3):
for k in range(0, 3):
z.iloc[i, j, k] = y.iloc[i] * x.iloc[j, k]
print(z)
有人可以幫我一些示例代碼嗎?謝謝!
uj5u.com熱心網友回復:
試試這個:
df1 = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
df2 = np.array([1, 2, 3, 4, 5])
df3 = df1 * df2[:, None, None]
輸出:
>>> df3
array([[[ 1, 2, 3],
[ 4, 5, 6],
[ 7, 8, 9]],
[[ 2, 4, 6],
[ 8, 10, 12],
[14, 16, 18]],
[[ 3, 6, 9],
[12, 15, 18],
[21, 24, 27]],
[[ 4, 8, 12],
[16, 20, 24],
[28, 32, 36]],
[[ 5, 10, 15],
[20, 25, 30],
[35, 40, 45]]])
轉載請註明出處,本文鏈接:https://www.uj5u.com/caozuo/394105.html
