Pandas Multiindex的数据类型[重复]

Question

import pandas as pd
index = pd.MultiIndex.from_tuples([(1,2, None), (1,2,3)])
print(index.get_level_values(2))

给予

Int64Index([nan, 3.0], dtype='float64')

但是，我想得到

Int64Index([nan, 3], dtype='object')

也就是说，Pandas 不应该自动更改数据类型。我该怎么做？

Answer 1

我认为您可以通过astype 将级别值转换为string，通过. 和get 第一个值将其转换为str.split。然后新建MultiIndex：

#change multiindex
new = zip(index.get_level_values(0),
          index.get_level_values(1),
          index.get_level_values(2).astype(str).str.split('.').str.get(0))


new_index = pd.MultiIndex.from_tuples(new)
print (new_index)
MultiIndex(levels=[[1], [2], [u'3', u'nan']],
           labels=[[0, 0], [0, 0], [1, 0]])

print(new_index.get_level_values(2))
Index([u'nan', u'3'], dtype='object')