将数组添加到pandas dataframe

时间:2016-01-21 19:42:02

标签: python pandas dataframe

我有一个名为A的pandas数据框,其中一列名为" a":

  • 约会
  • 2016-01-19 3
  • 2016-01-20:1
  • 2016-01-21:2

我有一个看起来像这样的数组:[4,3,2]。我想将此数组插入到数据框中,并为新列指定名称b。我该怎么做?

预期产出:

  • date a b
  • 2016-01-19 3 4
  • 2016-01-20:1 3
  • 2016-01-21:2 2

1 个答案:

答案 0 :(得分:3)

正如@mgc在评论中指出的那样,你可以做df['b'] = l

import pandas as pd
from io import StringIO
data="""
    date a
    2016-01-19 3
    2016-01-20 1
    2016-01-21 2
    """
df = pd.read_csv(StringIO(data), sep='\s+')

df = df.set_index('date')
df.index = pd.to_datetime(df.index)

print(df)
            a
date         
2016-01-19  3
2016-01-20  1
2016-01-21  2

l = [4,3,2]

df['b'] = l

print(df)
            a  b
date            
2016-01-19  3  4
2016-01-20  1  3
2016-01-21  2  2