Python - 如何获取 CSV 文件中所有列的数据类型？

Question

我正在尝试从 CSV 文件中获取每一列的所有数据类型。
文件中没有关于数据类型的文档，手动检查将花费很长时间(它有 150 列)。

开始使用这种方法:

df = pd.read_csv('/tmp/file.csv')

>>> df.dtypes
a   int64
b   int64
c   object
d   float64

上述方法是否足够好，还是有更好的方法来确定数据类型？
另外 - 文件有 150 列。当我键入 df.types 时 - 我只能看到 15 列左右。如何全部查看？

Answer 1

根据文件的大小，您可以通过使用 pd.read_csv 的 nrows 参数仅读取前几行来节省一些时间。 :

df = pd.read_csv('/tmp/file.csv', nrows=25)

这仅在您确定可以从前 n 行正确推断出类型时才有用，因此请谨慎使用。

将数据(或其子集)加载到 DataFrame 后，您可以通过多种不同的方式查看类型，其中一些已经发布，但我将使用一个简单的方式分享另一个循环和 iteritems :

for name, dtype in df.dtypes.iteritems():
    print(name, dtype)

a int64
b float64
c object