Определите типы данных столбцов фрейма данных


153

Я использую R и загрузил данные в кадре данных с помощью read.csv(). Как определить тип данных каждого столбца во фрейме данных?


Программно (например, sapply(..., class))или интерактивно (например str(...))) или и то и другое? Как правило, это более масштабируемо, чтобы сделать это программно, тогда вы можете произвольно составить Filter(...)список для целых чисел, символов, факторов и т. Д. Или вы можете использовать, grep/greplчтобы вывести типы столбцов, names(...)если они следуют каким-либо соглашениям об именах
SMCI

@smci: я не спрашивал «программно» в своем первоначальном вопросе. Я не знаю, почему вы изменили бы всю природу моего вопроса.
stackoverflowuser2010

хорошо, откатился Это не изменило всю природу, это прояснило это в одном из двух направлений. Использование интерактивных подходов str(...)не масштабируемо и заканчивается на <100 столбцов.
августа

Ответы:


215

Лучше всего начать с использования ?str(). Чтобы изучить некоторые примеры, давайте сделаем некоторые данные:

set.seed(3221)  # this makes the example exactly reproducible
my.data <- data.frame(y=rnorm(5), 
                      x1=c(1:5), 
                      x2=c(TRUE, TRUE, FALSE, FALSE, FALSE),
                      X3=letters[1:5])

Решение @Wilmer E Henao H очень упрощено:

sapply(my.data, class)
        y        x1        x2        X3 
"numeric" "integer" "logical"  "factor" 

Использование str()дает вам эту информацию плюс дополнительные плюсы (такие как уровни ваших факторов и первые несколько значений каждой переменной):

str(my.data)
'data.frame':  5 obs. of  4 variables:
$ y : num  1.03 1.599 -0.818 0.872 -2.682
$ x1: int  1 2 3 4 5
$ x2: logi  TRUE TRUE FALSE FALSE FALSE
$ X3: Factor w/ 5 levels "a","b","c","d",..: 1 2 3 4 5

Подход @Gavin Simpson также оптимизирован, но предоставляет немного иную информацию, чем class():

sapply(my.data, typeof)
       y        x1        x2        X3 
"double" "integer" "logical" "integer"

Для получения дополнительной информации о class, typeofи о среднем ребенке, modeпосмотрите эту превосходную ветку SO: всестороннего обзора типов вещей в R. 'mode' и 'class' и 'typeof' недостаточно .


1
После использования R в течение нескольких месяцев я обнаружил, что str(dataframe)это самый быстрый способ быстро определить типы столбцов. Другие подходы требуют большего количества нажатий клавиш и не показывают столько информации, но они полезны, если типы данных столбца являются входными данными для других функций.
stackoverflowuser2010

Привет, когда я сделал то же самое с apply вместо apply, это не сработало
Dom Jo

@ DomJo, зачем ты используешь apply()? Это для матриц. Фрейм данных - это (особый вид) список.
gung - Восстановить Монику


18

Я бы предложил

sapply(foo, typeof)

если вам нужны фактические типы векторов во фрейме данных. class()это немного другой зверь.

Если вам не нужно получать эту информацию в виде вектора (то есть вам не нужно делать что-то еще программно позже), просто используйте str(foo).

В обоих случаях fooбудет заменено имя вашего фрейма данных.


7

Просто передайте ваш фрейм данных в следующую функцию:

data_types <- function(frame) {
  res <- lapply(frame, class)
  res_frame <- data.frame(unlist(res))
  barplot(table(res_frame), main="Data Types", col="steelblue", ylab="Number of Features")
}

создать график всех типов данных в вашем фрейме данных. Для набора данных радужной оболочки мы получаем следующее:

data_types(iris)

введите описание изображения здесь


5

Для небольших кадров данных:

library(tidyverse)

as_tibble(mtcars)

дает вам распечатку из df с типами данных

# A tibble: 32 x 11
     mpg   cyl  disp    hp  drat    wt  qsec    vs    am  gear  carb
 * <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl>
 1  21       6  160    110  3.9   2.62  16.5     0     1     4     4
 2  21       6  160    110  3.9   2.88  17.0     0     1     4     4
 3  22.8     4  108     93  3.85  2.32  18.6     1     1     4     1

Для больших фреймов данных:

glimpse(mtcars)

дает вам структурированное представление типов данных:

Observations: 32
Variables: 11
$ mpg  <dbl> 21.0, 21.0, 22.8, 21.4, 18.7, 18.1, 14.3, 24.4, 22.8, 19.2, 17.8, 16.4, 17....
$ cyl  <dbl> 6, 6, 4, 6, 8, 6, 8, 4, 4, 6, 6, 8, 8, 8, 8, 8, 8, 4, 4, 4, 4, 8, 8, 8, 8, ...
$ disp <dbl> 160.0, 160.0, 108.0, 258.0, 360.0, 225.0, 360.0, 146.7, 140.8, 167.6, 167.6...
$ hp   <dbl> 110, 110, 93, 110, 175, 105, 245, 62, 95, 123, 123, 180, 180, 180, 205, 215...
$ drat <dbl> 3.90, 3.90, 3.85, 3.08, 3.15, 2.76, 3.21, 3.69, 3.92, 3.92, 3.92, 3.07, 3.0...
$ wt   <dbl> 2.620, 2.875, 2.320, 3.215, 3.440, 3.460, 3.570, 3.190, 3.150, 3.440, 3.440...
$ qsec <dbl> 16.46, 17.02, 18.61, 19.44, 17.02, 20.22, 15.84, 20.00, 22.90, 18.30, 18.90...
$ vs   <dbl> 0, 0, 1, 1, 0, 1, 0, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 0, 0, 0, 0, ...
$ am   <dbl> 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 0, 0, 0, 0, 0, ...
$ gear <dbl> 4, 4, 4, 3, 3, 3, 3, 4, 4, 4, 4, 3, 3, 3, 3, 3, 3, 4, 4, 4, 3, 3, 3, 3, 3, ...
$ carb <dbl> 4, 4, 1, 1, 2, 1, 4, 2, 2, 4, 4, 3, 3, 3, 4, 4, 4, 1, 2, 1, 1, 2, 2, 4, 2, ...

Чтобы получить список типов данных столбцов (как сказано выше @Alexandre):

map(mtcars, class)

дает список типов данных:

$mpg
[1] "numeric"

$cyl
[1] "numeric"

$disp
[1] "numeric"

$hp
[1] "numeric"

Чтобы изменить тип данных столбца:

library(hablar)

mtcars %>% 
  convert(chr(mpg, am),
          int(carb))

преобразует столбцы mpgи amв символ и столбец carbв целое число:

# A tibble: 32 x 11
   mpg     cyl  disp    hp  drat    wt  qsec    vs am     gear  carb
   <chr> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <chr> <dbl> <int>
 1 21        6  160    110  3.9   2.62  16.5     0 1         4     4
 2 21        6  160    110  3.9   2.88  17.0     0 1         4     4
 3 22.8      4  108     93  3.85  2.32  18.6     1 1         4     1
 4 21.4      6  258    110  3.08  3.22  19.4     1 0         3     1

3

Поскольку это не было четко указано, я просто добавляю это:

Я искал способ создать таблицу, которая содержит количество вхождений всех типов данных .

Скажем, у нас есть data.frameдва числовых и один логический столбец

dta <- data.frame(a = c(1,2,3), 
                  b = c(4,5,6), 
                  c = c(TRUE, FALSE, TRUE))

Вы можете суммировать количество столбцов каждого типа данных с этим

table(unlist(lapply(dta, class)))
# logical numeric 
#       1       2 

Это очень удобно, если у вас много столбцов и вы хотите получить краткий обзор.

Чтобы отдать должное: это решение было вдохновлено ответом @Cybernetic .


2

Вот функция, которая является частью пакета helpRFunctions , которая будет возвращать список всех различных типов данных в вашем фрейме данных, а также имена конкретных переменных, связанных с этим типом.

install.package('devtools') # Only needed if you dont have this installed.
library(devtools)
install_github('adam-m-mcelhinney/helpRFunctions')
library(helpRFunctions)
my.data <- data.frame(y=rnorm(5), 
                  x1=c(1:5), 
                  x2=c(TRUE, TRUE, FALSE, FALSE, FALSE),
                  X3=letters[1:5])
t <- list.df.var.types(my.data)
t$factor
t$integer
t$logical
t$numeric

Вы могли бы тогда сделать что-то вроде var(my.data[t$numeric]).

Надеюсь, это полезно!


1
Стоит отметить, что под капотом это lapply(your_data, class)с небольшой дополнительной обработкой для форматирования.
Грегор Томас

1

Если вы импортируете CSV-файл как data.frame (а не матрицу), вы также можете использовать summary.default

summary.default(mtcars)

     Length Class  Mode   
mpg  32     -none- numeric
cyl  32     -none- numeric
disp 32     -none- numeric
hp   32     -none- numeric
drat 32     -none- numeric
wt   32     -none- numeric
qsec 32     -none- numeric
vs   32     -none- numeric
am   32     -none- numeric
gear 32     -none- numeric
carb 32     -none- numeric

Используя наш сайт, вы подтверждаете, что прочитали и поняли нашу Политику в отношении файлов cookie и Политику конфиденциальности.
Licensed under cc by-sa 3.0 with attribution required.