python - 如何为 Pandas DataFrame 指定输入数据类型

Question

我想将现有的 Python 列表转换为 Pandas DataFrame 对象。如何指定每一列的数据格式和定义索引列？

这是我的代码示例:

import pandas as pd

data = [[1444990457000286208, 0, 286],
       [1435233159000067840, 0, 68],
       [1431544002000055040, 1, 55]]
df = pd.DataFrame(data, columns=['time', 'value1', 'value2'])

在上面的示例中，我需要为现有列设置以下类型:

• 时间:datetime64[ns]
• 值 1: bool 值
• value2: 整数

另外时间列应该用作索引列。

默认情况下所有三列都是 int64，我找不到如何在 DataFrame 对象创建期间指定列类型。

谢谢!

Answer 1

value2 已经是正确的 dtype。

对于 time，您可以使用 to_datetime 转换为日期时间然后用 set_index 设置索引.

对于 value1，您可以使用 astype 转换为 bool .

df['time'] = pd.to_datetime(df['time'])
df = df.set_index('time')
df['value1'] = df['value1'].astype(bool)