Читать только выбранные столбцы - PullRequest
116 голосов
/ 26 апреля 2011

Может кто-нибудь подсказать, как читать только первые 6 месяцев (7 столбцов) для каждого года из приведенных ниже данных, например, используя read.table()?

Year   Jan  Feb  Mar  Apr  May  Jun  Jul  Aug  Sep  Oct  Nov  Dec   
2009   -41  -27  -25  -31  -31  -39  -25  -15  -30  -27  -21  -25
2010   -41  -27  -25  -31  -31  -39  -25  -15  -30  -27  -21  -25 
2011   -21  -27   -2   -6  -10  -32  -13  -12  -27  -30  -38  -29

Ответы [ 4 ]

151 голосов
/ 26 апреля 2011

Скажем, данные находятся в файле data.txt, вы можете использовать аргумент colClasses read.table(), чтобы пропустить столбцы.Здесь данные в первых 7 столбцах равны "integer", а для оставшихся 6 столбцов мы устанавливаем "NULL", указывая, что они должны быть пропущены

> read.table("data.txt", colClasses = c(rep("integer", 7), rep("NULL", 6)), 
+            header = TRUE)
  Year Jan Feb Mar Apr May Jun
1 2009 -41 -27 -25 -31 -31 -39
2 2010 -41 -27 -25 -31 -31 -39
3 2011 -21 -27  -2  -6 -10 -32

Измените "integer" на один из принятых типов, как подробно описано в?read.table в зависимости от реального типа данных.

data.txt выглядит следующим образом:

$ cat data.txt 
"Year" "Jan" "Feb" "Mar" "Apr" "May" "Jun" "Jul" "Aug" "Sep" "Oct" "Nov" "Dec"
2009 -41 -27 -25 -31 -31 -39 -25 -15 -30 -27 -21 -25
2010 -41 -27 -25 -31 -31 -39 -25 -15 -30 -27 -21 -25
2011 -21 -27 -2 -6 -10 -32 -13 -12 -27 -30 -38 -29

и был создан с помощью

write.table(dat, file = "data.txt", row.names = FALSE)

, где dat is

dat <- structure(list(Year = 2009:2011, Jan = c(-41L, -41L, -21L), Feb = c(-27L, 
-27L, -27L), Mar = c(-25L, -25L, -2L), Apr = c(-31L, -31L, -6L
), May = c(-31L, -31L, -10L), Jun = c(-39L, -39L, -32L), Jul = c(-25L, 
-25L, -13L), Aug = c(-15L, -15L, -12L), Sep = c(-30L, -30L, -27L
), Oct = c(-27L, -27L, -30L), Nov = c(-21L, -21L, -38L), Dec = c(-25L, 
-25L, -29L)), .Names = c("Year", "Jan", "Feb", "Mar", "Apr", 
"May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"), class = "data.frame",
row.names = c(NA, -3L))

Если количество столбцов заранее неизвестно, служебная функция count.fields прочитает файл и посчитает количество полей в каждой строке.

## returns a vector equal to the number of lines in the file
count.fields("data.txt", sep = "\t")
## returns the maximum to set colClasses
max(count.fields("data.txt", sep = "\t"))
57 голосов
/ 18 октября 2015

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

1) С fread из пакета data.table:

Вы можете указать нужные столбцы с помощью параметра select из fread из пакета data.table. Вы можете указать столбцы с вектором имен столбцов или номеров столбцов.

Для примера набора данных:

library(data.table)
dat <- fread("data.txt", select = c("Year","Jan","Feb","Mar","Apr","May","Jun"))
dat <- fread("data.txt", select = c(1:7))

В качестве альтернативы, вы можете использовать параметр drop, чтобы указать, какие столбцы не следует читать:

dat <- fread("data.txt", drop = c("Jul","Aug","Sep","Oct","Nov","Dec"))
dat <- fread("data.txt", drop = c(8:13))

Все результаты в:

> data
  Year Jan Feb Mar Apr May Jun
1 2009 -41 -27 -25 -31 -31 -39
2 2010 -41 -27 -25 -31 -31 -39
3 2011 -21 -27  -2  -6 -10 -32

ОБНОВЛЕНИЕ: Если вы не хотите, чтобы fread возвращал data.table , используйте параметр data.table = FALSE, например: fread("data.txt", select = c(1:7), data.table = FALSE)

2) С read.csv.sql из пакета sqldf:

Другой альтернативой является функция read.csv.sql из пакета sqldf:

library(sqldf)
dat <- read.csv.sql("data.txt",
                    sql = "select Year,Jan,Feb,Mar,Apr,May,Jun from file",
                    sep = "\t")

3) С read_* -функциями из readr -пакета:

library(readr)
dat <- read_table("data.txt",
                  col_types = cols_only(Year = 'i', Jan = 'i', Feb = 'i', Mar = 'i',
                                        Apr = 'i', May = 'i', Jun = 'i'))
dat <- read_table("data.txt",
                  col_types = list(Jul = col_skip(), Aug = col_skip(), Sep = col_skip(),
                                   Oct = col_skip(), Nov = col_skip(), Dec = col_skip()))
dat <- read_table("data.txt", col_types = 'iiiiiii______')

Из документации объяснение используемых символов с col_types:

каждый символ представляет один столбец: c = символ, i = целое число, n = число, d = двойной, l = логический, D = дата, T = дата и время, t = время,? = угадай или _ / - пропусти столбец

7 голосов
/ 15 марта 2014

Вы также можете использовать JDBC для достижения этой цели. Давайте создадим пример CSV-файла.

write.table(x=mtcars, file="mtcars.csv", sep=",", row.names=F, col.names=T) # create example csv file

Загрузите и сохраните драйвер CSV JDBC по этой ссылке: http://sourceforge.net/projects/csvjdbc/files/latest/download

> library(RJDBC)

> path.to.jdbc.driver <- "jdbc//csvjdbc-1.0-18.jar"
> drv <- JDBC("org.relique.jdbc.csv.CsvDriver", path.to.jdbc.driver)
> conn <- dbConnect(drv, sprintf("jdbc:relique:csv:%s", getwd()))

> head(dbGetQuery(conn, "select * from mtcars"), 3)
   mpg cyl disp  hp drat    wt  qsec vs am gear carb
1   21   6  160 110  3.9  2.62 16.46  0  1    4    4
2   21   6  160 110  3.9 2.875 17.02  0  1    4    4
3 22.8   4  108  93 3.85  2.32 18.61  1  1    4    1

> head(dbGetQuery(conn, "select mpg, gear from mtcars"), 3)
   MPG GEAR
1   21    4
2   21    4
3 22.8    4
0 голосов
/ 22 мая 2019

Простой способ:

data <- read.table("dataname.csv", header = TRUE, sep = ",")[,1:7] 
...