我有一個 numpy 二維陣列,讓我們說
a = np.arange(1,7).reshape(2,3)
array([[1, 2, 3],
[4, 5, 6]])
我想要一個陣列,該陣列的每個元素沿一個軸的位置如下所示:
array([[0, 1, 2],
[0, 1, 2]])
我需要這個作為matplotlib.pyplot's的引數bar3d。有沒有一種簡單的方法可以做到這一點?
uj5u.com熱心網友回復:
使用np.tile:
import numpy as np
a = np.arange(1,7).reshape(2,3)
n_rows, n_cols = a.shape
res = np.tile(np.arange(n_cols), (n_rows, 1))
print(res)
輸出
[[0 1 2]
[0 1 2]]
轉載請註明出處,本文鏈接:https://www.uj5u.com/qiye/355586.html
上一篇:使用遞回函式查找陣列的最小數
下一篇:檢測到堆損壞(動態陣列)
