熊貓 - 只爲某些值獲取假人

Question

我有一個Pandas系列10000行，填充了單個字母表，從A到Z. 但是，我想創建僅用於A，B和C，使用熊貓get_dummies。 我該如何去做呢？

我不想獲取列中所有行值的虛擬變量，然後選擇特定列，因爲列中包含最終導致內存錯誤的其他冗餘數據。

Answer 1

試試這個：

# create mock dataframe 
df = pd.DataFrame({'alpha':['a','a','b','b','c','e','f','g']}) 

# use replace with a regex to set characters d-z to None 
pd.get_dummies(df.replace({'[^a-c]':None},regex =True)) 

輸出：

alpha_a alpha_b  alpha_c 
0 1 0 0 
1 1 0 0 
2 0 1 0 
3 0 1 0 
4 0 0 1 
5 0 0 0 
6 0 0 0 
7 0 0 0