python - 如何将 pandas DataFrame 的索引 dtype 更改为 int32？

Question

DataFrame 索引的默认 dtype 是 int64，我想将其更改为 int32。

我尝试用 pd.DataFrame.set_index 改变它和 int32 的 NumPy 数组，也尝试使用 dtype=np.int32 创建新索引。它不起作用，总是返回 int64 的索引。

有人可以展示一个工作代码来生成 int32 大小的 Pandas 索引吗？

我使用的是 conda Pandas v0.20.1。

Answer 1

不确定这在实践中是否值得做，但以下应该可行:

class Int32Index(pd.Int64Index):
    _default_dtype = np.int32

    @property
    def asi8(self):
        return self.values

i = Int32Index(np.array([...], dtype='int32'))

(来自 here)