python-从数据框中获取列数据类型

Question

我正在尝试从数据框中获取列数据类型

这是一个示例代码：

    print training_data.schema
    print 'fields'
    print training_data.schema.fields
    print 'names'
    print training_data.schema.names

上面的代码打印如下： StructType(List(StructField(id,LongType,true),StructField(text,StringType,true),StructField(label,DoubleType,true))) 字段 [StructField(id,LongType,true), StructField(text,StringType,true), StructField(label,DoubleType,true)] 名字 ['id', '文本', '标签']

但是我怎样才能得到标签列的数据类型呢？非常感谢您的宝贵时间。

问候

Answer 1

以下是如何获取 pandas 数据框的列名和列类型的复制粘贴示例：

import pandas as pd

list = [['Tom',34, 45.5], ['Jack',23, 60.5]]
df = pd.DataFrame(list, columns=["Name","Age","Pay"])

for column in df.columns:
    print("Column ", column, "is dtype:", df[column].dtype.name)

结果：

Column  Name is dtype: object
Column  Age is dtype: int64
Column  Pay is dtype: float64

Answer 2

感谢所有回复。 我找到了以下解决方案，希望对任何寻找答案的人有所帮助：

       for f, v in zip(df.schema.fields, df.schema.names):
          if v == colname:
            datatype = f.dataType

Answer 3

df['col label'].dtype

是一种选择。

编辑

name_dtype = df['col label'].dtype.name