如何在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]