pandas 正则化匹配 行转列

pandas 正则化匹配 行转列

有如下一个题目：
df=pd.DataFrame({‘id’:[{1,2,3,},{5,6,7},{9,10,11},[{12},{13},{15}]],‘value’:[1,2,3,4]})
需实现如下功能：
id value
1 1
2 1
3 1
5 2
6 2
…

实现代码如下

df['id']= df['id'].apply(lambda x:str(x)).apply(lambda x:re.sub('{|}|\[|\]','',x))
df= df['id'].str.split(',',expand=True).stack().reset_index(level=1, drop=True).reset_index(drop=False)
df.rename(columns={0:'id'}, inplace = True)
print(df)

–
朋友提供了一种效率更高的方式解决这个问题

from itertools import chain 
tmp_df = pd.DataFrame()
df['id'] = df['id'].apply(lambda x: list(chain(*x)) 
if isinstance(x, list) else x)
tmp_df['id'] = list(chain(*df['id'])) 
tmp_df['value'] = list(chain(*[[j]*i for i,j in zip(df['id'].apply(len), df['value'])]))

---

版权归于Eshter，转载请告知
撰写by： Eshter
时间：20190802