在R中,从df中采样n行,其中某个列具有非NA值(有条件地采样)
背景
这是一个玩具df:
df <- data.frame(ID = c("a","b","c","d","e","f"),
gender = c("f","f","m","f","m","m"),
zip = c(48601,NA,29910,54220,NA,44663),stringsAsFactors=FALSE)
如您所见,我NA在zip列中有几个值。
问题
我正在尝试从中随机采样 2 整行df- 但我希望它们zip是不为空的行。
我试过的
这段代码为我提供了一个基本的(即无条件的)随机样本:
df2 <- df[sample(nrow(df), 2), ]
但是,当然,这只能让我实现目标的一半——很多时候它会返回一个NA值为 in的行zip。此代码尝试添加条件:
df2 <- df[sample(nrow(df$zip != NA), 2), ]
我想我已经接近了,但这会产生错误invalid first argument。
有任何想法吗?
回答
我们可以用 is.na
tmp <- df[!is.na(df$zip),]
> tmp[sample(nrow(tmp), 2),]
回答
我们可以使用rownames+na.omit对行进行采样
> df[sample(rownames(na.omit(df["zip"])), 2),]
ID gender zip
3 c m 29910
4 d f 54220
回答
这是一个基本的 R 解决方案 complete.cases()
# define a logical vector to identify NA
x <- complete.cases(df)
# subset only not NA values
df_no_na <- df[x,]
# do the sample
df_no_na[sample(nrow(df_no_na), 2),]
输出:
ID gender zip
3 c m 29910
6 f m 44663