1

我有两个数组，我正在尝试使用连接进行组合：

a = np.array(([1,2], [5,6], [9,10]))
b = np.array(([3,4], [7,8], [11,12], [13,14], [17,18]))
c = np.concatenate((a,b), 1)

这不起作用，因为数组的长度不同。因此，我使用 len 来比较两个数组的长度，然后根据最小长度确定 c 的长度：

alength = len(a)
blength = len(b)
lengthforc = min(alength, blength)

在此示例中，最小值lengthforc为 3。因此，我尝试使用 if 语句通过删除最后两行（元素）来减少 b 的长度。

if blength > lengthforc:
   rowstoremove = blength - lengthforc

我如何修改这个 if 语句来执行我想要的（因为要删除的行数会改变），除非有另一种方法？最终的数组应该是：

>>> print c
  [[ 1  2  3  4]
  [ 5  6  7  8]
  [ 9 10 11 12]]
4

2 回答 2

1

尝试hstack：

a = np.array(([1,2], [5,6], [9,10]))
b = np.array(([3,4], [7,8], [11,12], [13,14], [17,18]))
end = min(a.shape[0], b.shape[0])
np.hstack((a[:end], b[:end]))

结果：

array([[ 1,  2,  3,  4],
       [ 5,  6,  7,  8],
       [ 9, 10, 11, 12]])

编辑

如果您不需要 NumPy 数组，您可以使用列表：

a_list = [[1,2], [5,6], [9,10]]
b_list = [[3,4], [7,8], [11,12], [13,14], [17,18]]

只有一行：

[x + y for x, y in zip(a_list, b_list)]

结果：

[[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]]
于 2013-06-02T15:12:03.780 回答
0

我不确定我是否正确地解决了问题（在我的解决方案中我只使用了列表），但就这么简单吗？

a = [[1,2], [5,6], [9,10]]
b = [[3,4], [7,8], [11,12], [13,14], [17,18]]
c = []
for x,y in zip(a,b):
    c.append(x+y)
于 2013-06-02T15:04:07.247 回答