web-dev-qa-db-ja.com

csvファイルから列のマルチインデックスを読み取る

次のような.csvファイルがあります。

Male, Male, Male, Female, Female
R, R, L, R, R
.86, .67, .88, .78, .81

私はそれをdfに読みたいので、私は持っています:

    Male        Female
    R       L   R
0   .86 .67 .88 .78 .81

やった:

df = pd.read_csv('file.csv', header=[0,1])

しかし、headersはそれをカットしません。その結果

Empty DataFrame
Columns: [(Male, R), (Male, R), (Male, L), (Female, R), (Female, R)]
Index: []

しかし、ヘッダーのドキュメントには次のように書かれています:

(...)Can be a list of integers that specify row
locations for a multi-index on the columns E.g. [0,1,3]

何が悪いのですか?どうすれば機能させることができますか?

22
nutship

問題は、列が重複していることだと思います:2(女性、R)。

バグなのか、列の重複が許容できないのかはわかりません。これはあなたのための回避策です:

まず、tupleize_cols = Trueでcsvを読み取ります

In [61]: df = pd.read_csv('test.csv', header=[0, 1], skipinitialspace=True, tupleize_cols=True)

In [62]: df
Out[62]: 
   (Male, R)  (Male, R)  (Male, L)  (Female, R)  (Female, R)
0       0.67       0.67       0.88         0.81         0.81

[1 rows x 5 columns]

次に、列のタイプをインデックスからマルチインデックスに変換します

In [63]: df.columns = pd.MultiIndex.from_tuples(df.columns)

In [64]: df
Out[64]: 
   Male              Female      
      R     R     L       R     R
0  0.67  0.67  0.88    0.81  0.81

[1 rows x 5 columns]
29
waitingkuo

パンダのバージョン0.21以降、MultiIndexesはデフォルトで作成されるので、df = pd.read_csv('file.csv', header=[0,1])が機能します。

10