如何在Python中创建相同大小的数组

时间:2017-01-24 12:53:06

标签: python arrays concatenation

我想知道有一个" easy"创建两个匹配数组的方法,在两个数组中插入一些虚拟缺失值,使它们保持相同的大小,两个数组中相同的索引保持不变,例如:

["A", "B", "C", "D", "E", "F"]["B", "C", "E"]

将是

["A", "B", "C", "D", "E", "F"]["N/A", "B", "C", "N/A", "E", "N/A"]

提前致谢: - )

2 个答案:

答案 0 :(得分:7)

列表理解中的单行代码:

array_1 = ["A", "B", "C", "D", "E", "F"]
array_2 = {"B", "C", "E"}

array_3 = [x if x in array_2 else "N/A" for x in array_1]

print(array_3)

结果:

['N/A', 'B', 'C', 'N/A', 'E', 'N/A']

请注意,我将array_2转换为set以便更快地查找。

答案 1 :(得分:3)

array_1 = ["A", "B", "C", "D", "E", "F"] 
array_2 = ["B", "C", "E"]

array_3 = array_1

for n,x in enumerate(array_3):

    if x not in array_2:

        array_3[n] = np.nan

print (array_3)

出:

[nan, 'B', 'C', nan, 'E', nan]