PY-20932 Pandas DataFrame viewer for multiindex frames raises TypeError
[idea/community.git] / python / testData / debug / test_dataframe_multiindex.py
1 import pandas as pd
2 import numpy as np
3
4 # multiindex rows
5 frame1 = pd.DataFrame(data=np.random.randint(0, high=10, size=(4, 2)), columns=['a', 'b'], index=pd.MultiIndex([['s', 'd'], [2, 3]], [[0, 0, 1, 1], [0, 1, 0, 1]]))
6 print(frame1) #line 6
7
8 # multiindex columns
9 frame2 = pd.DataFrame(np.random.random((4, 4)))
10 frame2.columns = pd.MultiIndex.from_product([[1, 2], [1, 'B']])
11 print(frame2) # line 11