Соединение 4 Pandas фреймов данных и установка 2 столбцов индекса - PullRequest
Новая
       25

Соединение 4 Pandas фреймов данных и установка 2 столбцов индекса

0 голосов
/ 29 февраля 2020

Простите, если этот вопрос кажется дублирующим, но я не смог найти предыдущий вопрос, который точно соответствует моим требованиям.

У меня есть фрейм данных с именем "df", который имеет индекс datetime и выглядит как this: 

            X1  X2  X3  X4
date                
1/1/2000    10  4   1   6
1/2/2000    9   8   7   2
1/3/2000    7   10  5   3
1/4/2000    10  6   2   7

Из кадра данных я произвел 3-кадровые данные процентных изменений переменных через 1, 2 и 3 дня: 

pct_1 = df.pct_change(1)
pct_2 = df.pct_change(2)
pct_3 = df.pct_change(3)

Кадр "pct_1" выглядит следующим образом : 

             X1       X2       X3       X4
date                
1/1/2000     NaN      NaN      NaN      NaN
1/2/2000    -0.10     1.00     6.00    -0.67
1/3/2000    -0.22     0.25    -0.29     0.50
1/4/2000     0.43    -0.40    -0.60     1.33

Кадр "pct_2" выглядит следующим образом: 

            X1      X2      X3      X4
date                
1/1/2000    NaN     NaN     NaN     NaN
1/2/2000    NaN     NaN     NaN     NaN
1/3/2000   -0.30    1.50    4.00   -0.50
1/4/2000    0.11   -0.25   -0.71    2.50

Кадр "pct_3" выглядит следующим образом: 

            X1  X2  X3   X4
date                
1/1/2000    NaN NaN NaN NaN
1/2/2000    NaN NaN NaN NaN
1/3/2000    NaN NaN NaN NaN
1/4/2000    0.0 0.5 1.0 0.166667

У меня есть другой фрейм данных , называемый «регионами», который выглядит следующим образом: 

Region  Priority
X1         1
X2         2
X3         3
X4         1

Моя конечная цель - создать что-то похожее на это: 

                        pct_1   pct_2   pct_3   priority
date    region              
1/1/2000    X1          NaN     NaN     NaN     1
1/2/2000    X2          1.00    NaN     NaN     2
1/3/2000    X3         -0.29    4.00    NaN     3
1/4/2000    X4          1.33    2.50    0.17    1

Кто-нибудь знает, как это может быть сделанный? Я понятия не имею о логи c или что-нибудь. Я просто знаю, как должен выглядеть конечный продукт.

1 Ответ

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

Вычисление фреймов данных pct_1, pct_2, pct_3 прямолинейно, но при объединении их необходимо переформатировать данные. Извлечение Pandas -реформирование

Предполагается, что столбец Region имеет уникальные значения. 

str1 = r'''
date        X1  X2  X3  X4
1/1/2000    10  4   1   6
1/2/2000    9   8   7   2
1/3/2000    7   10  5   3
1/4/2000    10  6   2   7'''

str2=r'''
Region  Priority
X1         1
X2         2
X3         3
X4         1
'''

reg = pd.read_table(io.StringIO(str2), sep='\\s+')

# Read data
df = pd.read_table(io.StringIO(str1), sep='\\s+', index_col=0)

# Compute the percent change and stack the Region column
pct = pd.concat([df.pct_change(i) for i in range(1,4)], axis=1, keys=['p1', 'p2', 'p3'])
pct = pct.stack()

pct.index = pct.index.set_names(['date', 'Region'])
pct = pct.reset_index()

# Reformatting
pct = pd.merge(pct, reg, on='Region')
pct = pct.set_index(['date', 'Region'])

print(pct.to_string())
...