如何向 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中文网其他相关文章！