2016-10-03 66 views
3

Ive得到不同長度的兩個向量: -如何向量化在python陣列

a=[1,2,4,6] 
b=[2,3,4,8,9] 

基於這些矢量我想在輸出一個1/0的索引作爲

[1,1,0,1,0,1,0,0,0,0] 
[0,1,1,1,0,0,0,1,1,0] 

任何人都可以幫忙嗎?

回答

2

一個簡單的解決辦法是:

>>> a=[1,2,4,6] 
>>> b=[2,3,4,8,9] 
>>> 
>>> [int(i in a) for i in range(1,11)] 
[1, 1, 0, 1, 0, 1, 0, 0, 0, 0] 
>>> [int(i in b) for i in range(1,11)] 
[0, 1, 1, 1, 0, 0, 0, 1, 1, 0]