numpy的stack(),hstack()函数

stack(value, axis)

通过axis=0或1，对value进行堆积

• 当axis=0时

a = [1, 2, 3, 4]
b = [5, 6, 7, 8]
d = np.stack((a, b),axis=0)
print(d)
print(d[1, 1])

输出：d ==> [[1, 2, 3, 4], [5, 6, 7, 8]]
d[1, 1] ==> 6

---

• 当axis=1时

a = [1, 2, 3, 4]
b = [5, 6, 7, 8]
d = np.stack((a,b),axis=1)
print(d)
print(d[1,1])

输出：d ==> [[1, 5],[2, 6],[3, 7],[4, 8]]
d[1, 1] ==> 6

• numpy中list的查找格式为list_one[row, column]；如果写成list[row, :]代表取row那一行
• python中list的查找格式为list[row][column]

---

hstack(value)

value必须具有相同的数据结构，数据类型不限，可以是python的列表或者元祖，或者是numpy列表，hstack会将多个value(value_one, value_two)的相同维度的数值组合在一起，并以同value同样的数据结构返回numpy数组

a=(1, 2, 3)
b=(4, 5, 6)
print(np.hstack((a, b)))

输出：[1, 2, 3, 4, 5, 6]

a=[[1], [2], [3]]
b=[[1], [2], [3]]
print(np.hstack((a, b)))

输出：[[1, 1], [2, 2], [3, 3]]

---

更加详细说明