python - получить столбец dataType из кадра данных - PullRequest
Новая
       12

python - получить столбец dataType из кадра данных

0 голосов
/ 02 июля 2018

Я пытаюсь получить тип данных столбца из кадра данных

Вот пример кода: 

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

Код выше печатается, как показано ниже: StructType (List (StructField (идентификатор, LongType, правда), StructField (текст, StringType, правда), StructField (этикетка, DoubleType, правда))) поля [StructField (id, LongType, true), StructField (текст, StringType, true), StructField (метка, DoubleType, true)] имена ['id', 'text', 'label']

Но как я могу получить тип данных для столбца метки? Большое спасибо за ваше время.

Привет

Ответы [ 2 ]

0 голосов
/ 02 июля 2018

Спасибо за все ответы. Я нашел приведенное ниже решение, надеясь, что оно будет полезно любому, кто ищет ответ: 

       for f, v in zip(df.schema.fields, df.schema.names):
          if v == colname:
            datatype = f.dataType
0 голосов
/ 02 июля 2018 
df['col label'].dtype

Есть один вариант.

Редактировать 

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