Python中Numpy数组for循环拼接仅获最后一行问题求助

嘿，我明白你遇到的问题了——肯定是在循环里每次都直接把新行赋值给了结果数组，而不是把它追加进去，所以最后只留下了最后一次循环的内容对吧？别担心，我给你两种解决方案，一种是修复循环的写法，另一种是更适合numpy的高效操作：

方法1：修复循环追加逻辑 #

如果一定要用for循环的话，你需要先初始化一个空列表，然后在每次循环里把取出的行追加到列表里，最后再转成numpy数组。这样就能保留所有行啦：

import numpy as np

# Array inputs
a = np.array([0, 1, 2])
b = np.array([[1, 10, 20, 30], [2, 40, 50, 60], [3, 70, 80, 90]])

# 初始化空列表用来存结果
result = []
for idx in a:
    # 把对应行追加到列表里，而不是直接赋值
    result.append(b[idx])

# 转成numpy数组
result_array = np.array(result)
print(result_array)

运行后你会得到包含所有行的数组：

[[ 1 10 20 30]
 [ 2 40 50 60]
 [ 3 70 80 90]]

如果a是[1,1,2]这种重复的情况，结果也会自动重复对应的行：

[[ 2 40 50 60]
 [ 2 40 50 60]
 [ 3 70 80 90]]

方法2：用numpy的数组索引（更高效） #

其实numpy本身就支持整数数组索引，完全不用写循环，一行代码就能搞定，而且效率比循环高很多（尤其是数据量大的时候）：

import numpy as np

a = np.array([0, 1, 2])
b = np.array([[1, 10, 20, 30], [2, 40, 50, 60], [3, 70, 80, 90]])

# 直接用a作为索引取出b中对应的行
result_array = b[a]
print(result_array)

这个方法同样支持a有重复元素的情况，结果和循环方法完全一致，但代码更简洁，运行更快。

额外说明 #

如果a里的元素不是b的行索引，而是要匹配b每行的第一个元素（比如a里的1对应b中第一个元素为1的行），那你可以先找到对应的索引再提取：

# 假设要匹配b每行的第一个元素
indices = np.where(b[:, 0] == a[:, None])[1]
result_array = b[indices]

不过看你给出的示例，a是[0,1,2]对应b的3行，应该是用行索引的场景，所以前面两种方法就够用啦。

内容的提问来源于stack exchange，提问作者Adam