Кросс-табуляция между двумя наборами столбцов в DataFrame - PullRequest
Новая
       37

Кросс-табуляция между двумя наборами столбцов в DataFrame

0 голосов
/ 17 января 2019

Я попытался получить перекрестную таблицу данных, указанных срезом.

Но что-то не так в синтаксисе.

data.csv как показано ниже 

ia,ib,ic,id,ie,if,ig  
a,0,0,0,e,0,g         
0,b,0,0,e,f,0         
0,0,c,d,0,f,g

А затем выполните python3 test.py, как показано ниже 

import pandas as pd
import enum

df = pd.read_csv('data.csv')

class Slices(enum.Enum):
    first = slice(0, 2)
    second = slice(4, 6)

def getCrosstab(*args):
    cols1 = []
    cols1.append(df.iloc[:, args[0].value])
    cols2 = []
    cols2.append(df.iloc[:, args[1].value])
    print( pd.crosstab(cols1, cols2) )

if __name__ == '__main__':
    getCrosstab(Slices.first, Slices.second)

Ожидаемый результат: 

col2 ie if ig
col1
ia   1  0  1
ib   1  1  0
ic   0  1  1

Но у меня была ошибка:
ValueError: Форма переданных значений (2, 2), индексы подразумевают (2, 3)

Я не могу полностью понять значение этой ошибки.

Пожалуйста, дайте мне ваше руководство.

1 Ответ

0 голосов
/ 17 января 2019

melt дважды, один раз для каждого набора столбцов, а затем вызовите crosstab: 

u = (df.melt(['ia', 'ib', 'ic'], var_name='C', value_name='D')
       .melt(['C', 'D'], var_name='A', value_name='B')
       .query("B != '0' and D != '0'"))

pd.crosstab(u.A, u.C)

C   id  ie  if  ig
A                 
ia   0   1   0   1
ib   0   1   1   0
ic   1   0   1   1

def crosstab_for(df, sliceA, sliceB):
    u = (df.reindex(df.columns[sliceA] | df.columns[sliceB], axis=1)
           .melt(df.columns[sliceA], var_name='C', value_name='D')
           .melt(['C', 'D'], var_name='A', value_name='B')
           .query("B != '0' and D != '0'"))

    return pd.crosstab(u.A, u.C)

crosstab_for(df, slice(0, 3), slice(4, 7))

C   ie  if  ig
A             
ia   1   0   1
ib   1   1   0
ic   0   1   1
...