如何在Python中制作相同大小的数组

Question

我想知道是否有一种“简便”的方法来创建两个匹配的数组，从而在两个数组中插入一些虚拟的缺失值，以便它们保持相同的大小，并且两个数组中相同的索引保持相同，因此例如：

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

将会

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

提前致谢 ：-）

Answer 1

列表理解中的一线实现：

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以便更快地查找。

Answer 2

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]