我是 Python 新手。我有一個 2D np.array（例如 50 行和 12 列），當第一列==x 和第九列==y 時，我需要第三列的平均值。如果不使用ifs，我無法弄清楚如何做到這一點......任何幫助，將不勝感激。
3 回答
白板的微信
TA貢獻1883條經驗 獲得超3個贊
假設您的數組被稱為arr。在這種情況下，您希望先應用兩個不同的過濾器，然后再應用1st column==x第二個9th column==y。首先，您應該單獨創建每個過濾器（掩碼），然后根據它們與預期輸出之間的邏輯關系來查看您想對它們做什么。
mask1 = arr[:, 0] == x # 1st column==x
mask1 = arr[:, 8] == y # 9th column==y
現在，您可以使用or、and或任何其他邏輯運算符來創建最終掩碼，在本例中為and. 為此，您可以在 numpy 中使用邏輯函數。
final_mask = np.logical_and(mask1, mask2)
最后，您需要final_mask做的就是根據您要執行的計算過濾您的數組：
filtered_3rd_column = arr[:, final_mask]
_mean = filtered_3rd_column.mean()
鳳凰求蠱
TA貢獻1825條經驗 獲得超4個贊
一只萌萌小番薯
TA貢獻1795條經驗 獲得超7個贊
我解決了如下問題（感謝 Kasramvd）：
mask1 = arr[:, 0] == x # 1st column==x
mask1 = arr[:, 8] == y # 9th column==y
final_mask = np.logical_and(mask1, mask2)
filtered_arr = arr[final_mask,:]
mean_3rd_column = filtered_arr[:,2].mean()
添加回答
舉報
0/150
提交
取消