使用生成器表达式将List的List转换为DataFrame

时间:2017-12-31 08:22:07

标签: python pandas dataframe

s 是一个Series对象:

>>> s
0     [2010, 1]
1     [2011, 5]
2    [2012, 10]
dtype: object

然后我遇到了以下内容,它使用生成器表达式将 s 转换为DataFrame。

>>> df = pd.DataFrame(i for i in s)
>>> df
      0   1
0  2010   1
1  2011   5
2  2012  10

有没有解释为什么这有效?我不明白为什么会这样。

2 个答案:

答案 0 :(得分:2)

这就是它工作的原因。在场景后面,Generator将转换回list。这是source code的一大块。

if isinstance(data, types.GeneratorType):
  data = list(data)

它与list的作用相同。

# Creating a list
l = [1,2,3]
# Using the generator
df1 = pd.DataFrame(i for i in l)
# Using the list
df2 = pd.DataFrame(l)
df1.equals(df2)
# True

答案 1 :(得分:0)

我认为这是最直接的解决方案:

df = pd.DataFrame({'a': [[2010, 1],[2011, 5],[2012, 10]]})

df[['a1', 'a2']] = df['a'].apply(pd.Series)