地圖熊貓系列列出會員

說我有以下幾大系列，dtype('O')：地圖熊貓系列列出會員

s = pd.Series(list('abcde'))

而下面的列表（組）：

group1 = list('ab') 
group2 = list('cd') 
group3 = list('xy')

我要地圖s到一個新的系列根據組成員資格，在沒有成員資格的情況下填寫np.nan。結果：

res 
Out[17]: 
0 group1 
1 group1 
2 group2 
3 group2 
4  nan 
dtype: object

我在尋找這樣做不是使用np.where重複一個更Python的方式：

res = pd.Series(np.where(s.isin(group1), 'group1', 
         np.where(s.isin(group2), 'group2', 
         np.where(s.isin(group3), 'group3', 
         np.nan))))

理想的情況下，這將意味着做這樣的事情，指定列表的集合作爲一個單一可迭代的，因爲在我真正的問題中，我有更多的組。 res可以是dtype O或categorical。

我已經試過什麼：

上面的例子有np.where，這工作，但過於冗長。
創建dict(zip(['group1', 'group2', 'group3'], [group1, group2, group3]))。卡在那裏。

我有一種感覺，這可能是重複的，但我第一次嘗試時無法找到。

注意：組列表將永遠是不相交的。

來源

2017-07-10 Brad Solomon

我之前沒有機會回覆您的評論。我認爲你最好的選擇是在開始時構建映射，但我不知道你是如何得到這些列表的。無論如何，如果效率是一個問題，請不要接受。如果問題沒有結束，人們更有可能從事替代方案。 – ayhan

我認爲你需要重構映射。

從字典（dict(zip(['group1', 'group2', 'group3'], [group1, group2, group3]))）開始：

d = {'group1': list('ab'), 'group2': list('cd'), 'group3': list('xy')}

的逆轉之一，假設不相交的名單，將是：

reversed_map = {v: k for k, lst in d.items() for v in lst}

現在，你可以使用地圖：

s.map(reversed_map) 
Out[19]: 
0 group1 
1 group1 
2 group2 
3 group2 
4  NaN 
dtype: object

來源

2017-07-10 12:47:25 ayhan

地圖熊貓系列列出會員

回答

相關問題