我有一個數(shù)據(jù)框和一個列表，其中包括數(shù)據(jù)框中的部分列名稱，如下所示： my_frame: col1, col2, col3, ..., coln 2, 3, 4, ..., 2 5, 8, 5, ..., 1 6, 1, 8, ..., 9 my_list: ['col1','col3','coln']現(xiàn)在，我想創(chuàng)建一個與原始數(shù)據(jù)幀（總列數(shù)）大小相同的數(shù)組，其中僅包含零和一?；旧?，如果“my_list”中有類似的列名稱，我希望數(shù)組包含 1，否則包含 0。我想要的輸出應該是這樣的： my_array={[1,0,1,0,0,...,1]}
1 回答
長風秋雁
TA貢獻1757條經(jīng)驗 獲得超7個贊
這應該可以幫助你：
import pandas as pd
dictt = {'a':[1,2,3],
'b':[4,5,6],
'c':[7,8,9]}
df = pd.DataFrame(dictt)
my_list = ['a','h','g','c']
my_array = []
for column in df.columns:
if column in my_list:
my_array.append(1)
else:
my_array.append(0)
print(my_array)
輸出：
[1, 0, 1]
如果你想使用my_arrayanumpy array而不是 a list，那么使用這個：
import pandas as pd
import numpy as np
dictt = {'a':[1,2,3],
'b':[4,5,6],
'c':[7,8,9]}
df = pd.DataFrame(dictt)
my_list = ['a','h','g','c']
my_array = np.empty(0,dtype = int)
for column in df.columns:
if column in my_list:
my_array = np.append(my_array,1)
else:
my_array = np.append(my_array,0)
print(my_array)
輸出：
[1 0 1]
為了更容易理解，我在代碼中使用了測試數(shù)據(jù)。您可以用您的實際數(shù)據(jù)替換測試數(shù)據(jù)（即用您的實際數(shù)據(jù)幀替換我的測試數(shù)據(jù)幀）。希望這有幫助！
添加回答
舉報
0/150
提交
取消