如何為 NumPy 陣列新增列？

如何使用附加列來增強 NumPy 陣列

使用 NumPy 陣列時，通常需要在陣列中新增額外的列數組。這可以使用多種方法來實現，其中之一是使用 NumPy 的 np.c_[...] 函數。

考慮以下2D 陣列：

<code class="python">a = np.array([
    [1, 2, 3],
    [2, 3, 4],
])</code>

加入列這個陣列的零，我們可以使用np.c_[...]:

<code class="python">b = np.c_[a, np.zeros(a.shape[0])]</code>

這將建立一個新的陣列b 附加一列零，結果為：

<code class="python">b = np.array([
    [1, 2, 3, 0],
    [2, 3, 4, 0],
])</code>

新增列的另一種替代方法是使用np.r_[...] 函數。它可用於向數組追加行和列，提供更大的靈活性。

例如，增加額外的一列：

<code class="python">c = np.c_[a, np.ones(a.shape[0])]</code>

增加多列：

<code class="python">d = np.c_[a, 2*np.ones(a.shape[0]), 3*np.ones(a.shape[0])]</code>

這些函數中的方括號表示法[...] 允許透過括號表示法[...] 允許透過在所需位置指定值或陣列來靈活新增列。

以上是如何為 NumPy 陣列新增列？的詳細內容。更多資訊請關注PHP中文網其他相關文章！