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