我有一個(gè)列表，其中“SUMMARY”元素和下一個(gè)元素之間的項(xiàng)目數(shù)不固定list = ['SUMMARY - Dec 2013', 'Person1', 'None', 'None', '10', 'SUMMARY - Dec 2013', 'Person2', '20', 'SUMMARY - Jan 2014', 'Person3', 'None']我想要實(shí)現(xiàn)的是將列表轉(zhuǎn)換為數(shù)據(jù)框，其中每一行都以“SUMMARY”元素開頭。我用過match = []match.append([n for n, l in enumerate(list) if l.startswith('SUMMARY')])使用輸出[[0, 5, 8]]獲取包含“SUMMARY”的項(xiàng)目的索引，我希望我的數(shù)據(jù)框的每一行都以其索引包含在match. 在這種情況下，匹配有 3 個(gè)元素，所以我希望我的數(shù)據(jù)框有 3 行和以下結(jié)構(gòu)：'SUMMARY - Dec 2013', 'Person1', 'None', 'None', '10''SUMMARY - Dec 2013', 'Person2', '20', NA, NA'SUMMARY - Jan 2014', 'Person3', 'None', NA, NA基本上，當(dāng)特定行的項(xiàng)目數(shù)小于最大列數(shù)時(shí)，其余部分將填充 NA/NaN。
1 回答
蕪湖不蕪
TA貢獻(xiàn)1796條經(jīng)驗(yàn) 獲得超7個(gè)贊
如果您從單個(gè)系列開始，則更容易解決這個(gè)問題。您可以根據(jù)“摘要”是否存在對數(shù)據(jù)行進(jìn)行分組，然后使用它來跨列聚合和重新分解數(shù)據(jù)。
s = pd.Series(your_list)
pd.DataFrame(s.groupby(s.str.contains('summary', case=False).cumsum())
.agg(list)
.tolist())
0 1 2 3 4
0 SUMMARY - Dec 2013 Person1 None None 10
1 SUMMARY - Dec 2013 Person2 20 None None
2 SUMMARY - Jan 2014 Person3 None None None
感謝@Shubham Sharma 提出迭代組的建議：
pd.DataFrame([g.tolist() for k, g in s.groupby(
s.str.contains('summary', case=False).cumsum())])
0 1 2 3 4
0 SUMMARY - Dec 2013 Person1 None None 10
1 SUMMARY - Dec 2013 Person2 20 None None
2 SUMMARY - Jan 2014 Person3 None None None
添加回答
舉報(bào)
0/150
提交
取消