将数组添加到pandas数据帧中

Question

I have a pandas dataframe called A with one column called "a":

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

I have one array that looks like: [4,3,2]. I want to insert this array into the dataframe and give the new column the name b. How do I do that?

Expected output:

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

Answer 1

As @mgc pointed out in the comment you could do 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