如何使用附加列来增强 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中文网其他相关文章!