简体   繁体   中英

Is there a way to concatenate a pandas Series/Datetime series with a numpy matrix of same rows?

I created a random matrix of given size using numpy. For a time series simulation, I created a time series with a frequency of a month for the corresponding matrix. Now I'd like to combine them and have them as a pandas dataframe. This is what I have so far -

import numpy as np
import pandas as pd

cols = ['time', 'cases', 'deaths', 'recoveries']

data = np.random.randint(0,50,(50,3))
times = pd.date_range('2019-12-01', periods=50, freq='MS')
df = pd.DataFrame(pd.concat(times, data, ignore_index=True), columns=cols)

This gives the following error at line 8 -

TypeError: cannot concatenate object of type '<class 'pandas._libs.tslibs.timestamps.Timestamp'>'; only Series and DataFrame objs are valid

So I tried converting it to series using times = pd.Series(pd.date_range('2019-12-01', periods=50, freq='MS')) however that in turn gave the error -

TypeError: first argument must be an iterable of pandas objects, you passed an object of type "Series"

Expected O/P -

|   time     |cases|deaths|recoveries|
|------------------------------------|
| 2019-12-01 | 0   | 0    | 0        |
| 2020-01-01 | 1   | 0    | 0        |
| 2020-02-01 | 2   | 1    | 0        |

I suggest create DatetimeIndex instead column, for possible processing by datetimelike methods of pandas:

#removed time column
cols = ['cases', 'deaths', 'recoveries']

data = np.random.randint(0,50,(50,3))
#added time in name parameter
times = pd.date_range('2019-12-01', periods=50, freq='MS', name='time')
#removed concat and added index parameter
df = pd.DataFrame(data, columns=cols, index=times)
print (df.head(10))
            cases  deaths  recoveries
time                                 
2019-12-01     28      44          25
2020-01-01     21      23          26
2020-02-01     15      17           5
2020-03-01     35       3          42
2020-04-01     46       7           3
2020-05-01     23      47          28
2020-06-01     31      30          34
2020-07-01      8       4          15
2020-08-01     46      14          24
2020-09-01     43      47           6

If need column only add DataFrame.reset_index :

df = pd.DataFrame(data, columns=cols, index=times).reset_index()
print (df.head(10))
        time  cases  deaths  recoveries
0 2019-12-01      2      26          43
1 2020-01-01     43      40          41
2 2020-02-01     23      12          22
3 2020-03-01     43      37          28
4 2020-04-01      7      26          20
5 2020-05-01     19      46          41
6 2020-06-01     43       1           0
7 2020-07-01     19      42           4
8 2020-08-01     14      39          40
9 2020-09-01     15       8          25

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM