Python多个数组合并

Python中多个数组合并为一个数组的方法整理。

extend方法

该方法可以扩展数组，会改变原始数组。

a = [1,2,3,4,7,5,6]
b = ['a','b']
c = ['h',12,'c']
a.extend(b)
a.extend(c)
print(a)

#结果：[1, 2, 3, 4, 7, 5, 6, 'a', 'b', 'h', 12, 'c']

直接相加

用+将各数组连接起来。

a = [1,2,3,4,7,5,6]
b = ['a','b']
c = ['h',12,'c']

d = a + b +c
print(d)

#结果：[1, 2, 3, 4, 7, 5, 6, 'a', 'b', 'h', 12, 'c']

flatten方法

flatten()方法是numpy中array数组的方法，使用时要导入包和类型转换。

from numpy import array

a = [1,2,3]
b = ['a','b','c']
c = ['h',12,'k']
e = [a,b,c]
e = array(e)
print(e.flatten())

#结果：['1' '2' '3' 'a' 'b' 'c' 'h' '12' 'k']

值得注意的是该方法不适用各数组中元素个数不同的情况。

a = [1,2,3,4]  #元素个数不同
b = ['a','b','c']
c = ['h',12,'k']
e = [a,b,c]
e = array(e)
print(e.flatten())

#结果：[list([1, 2, 3, 4]) list(['a', 'b', 'c']) list(['h', 12, 'k'])]

列表表达式

使用前得把多个数组放在一个数组中，应用在一个二维数组上。

a = [1,2,3,4] 
b = ['a','b','c']
c = ['h',12,'k']
e = [a,b,c]
e = [y for x in e for y in x]
print(e)

结果：[1, 2, 3, 4, 'a', 'b', 'c', 'h', 12, 'k']

更多