Есть ли способ объединить серии pandas Series / Datetime с numpy матрицей из тех же строк? - PullRequest
Новая
       6

Есть ли способ объединить серии pandas Series / Datetime с numpy матрицей из тех же строк?

0 голосов
/ 12 марта 2020

Я создал случайную матрицу заданного размера, используя numpy. Для моделирования временных рядов я создал временной ряд с периодичностью месяц для соответствующей матрицы. Теперь я хотел бы объединить их и сделать их pandas кадром данных. Это то, что у меня пока есть - 

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)

Это дает следующую ошибку в строке 8 - 

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

Так что я попытался преобразовать его в серию, используя times = pd.Series(pd.date_range('2019-12-01', periods=50, freq='MS')), однако, в свою очередь дал ошибку - 

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

Ожидаемый O / P - 

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

1 Ответ

1 голос
/ 12 марта 2020

Я предлагаю создать DatetimeIndex вместо столбца, для возможной обработки с использованием методов типа datetime, таких как 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

Если нужен только столбец, добавьте 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
...