时间戳

首页 > 解决方案 > 根据特定字符串提取列表元素并存储为熊猫数据框

python - 根据特定字符串提取列表元素并存储为熊猫数据框

问题描述

我有一个列表，其中包含如下所示的几点：

['Pros:', 'Pros 1', 'Pros 2','Cons:','Cons 1','Cons 2']

我正在尝试使用 2 列创建一个 Dataframe，一个称为Pros另一个称为Cons

预期输出：

Pros, Cons
Pros 1; Pros 2,Cons 1; Cons 2

我遇到的问题是Pros并且Cons在每个列表中都会有所不同，因此我试图将 Cons 之前的所有文本归类为一列，将 Cons 之后的所有项目归类为另一列

标签： pythonpandaslist

解决方案

我会采取另一种方法。用于zip_longest填充缺失值None。这是一种更好的pandas做法，将使呈现/处理数据更容易一些。

import pandas as pd
from itertools import zip_longest

li = ['Pros:', 'Pros 1', 'Pros 2', 'Cons:', 'Cons 1', 'Cons 2', 'Cons 3']
cons_index = li.index('Cons:')
pros, cons = li[1:cons_index], li[cons_index + 1:]
df = pd.DataFrame(zip_longest(pros, cons), columns=['Pros', 'Cons'])
print(df)

输出

     Pros    Cons
0  Pros 1  Cons 1
1  Pros 2  Cons 2
2    None  Cons 3

如果你坚持你的方法，那么

import pandas as pd

li = ['Pros:', 'Pros 1', 'Pros 2', 'Cons:', 'Cons 1', 'Cons 2', 'Cons 3']
cons_index = li.index('Cons:')
pros, cons = ';'.join(li[1:cons_index]), ';'.join(li[cons_index + 1:])
df = pd.DataFrame({'Pros': [pros], 'Cons': [cons]})

推荐阅读