R软件中的集合计算操作

Fri, 27 Sep 2013 00:00:00 +0000

在处理数据尤其是在编写函数或编译R包的时候会涉及到集合的运算操作。比如，判断某个元素是否属于某个集合，并用这个判断结果作为 if 结构中的条件语句。

在量化投资的编程中更是经常用到集合的运算操作。比如，判断某支股票是否在今日的投资组合里，又或者在已有当前投资组合的情况下如何生成交易单以得到理想投资组合的股票代码，这涉及到差集操作。因此，了解一下R中的集合计算操作是很有必要的。

根据小学时候的知识，我们知道集合计算无非包括这么几类：

它们对应的R函数就是下面这些：

本着解释知识不举例子就是耍流氓的精神，我们看看下面的例子:

# 随意生成两个向量x和y
x <- c(sort(sample(1:20, 9)), NA)
y <- c(sort(sample(3:23, 7)), NA)
# 查看 x，y 都长什么样子
x

## [1] 1 4 6 7 9 11 15 17 19 NA

## [1] 12 13 14 15 19 20 21 NA

# 计算并集
union(x, y)

## [1] 1 4 6 7 9 11 15 17 19 NA 12 13 14 20 21

# 计算交集
intersect(x, y)

## [1] 15 19 NA

# 计算差集，注意：计算差集时，集合的先后顺序是影响结果滴！
setdiff(x, y)

## [1] 1 4 6 7 9 11 17

setdiff(y, x)

## [1] 12 13 14 20 21

# x，y是否相等，当然不相等啦....
setequal(x, y)

## [1] FALSE

# 看看并集、交集、差集的关系
## True for all possible x & y :
setequal( union(x, y),
 c(setdiff(x, y), intersect(x, y), setdiff(y, x)))

## [1] TRUE

# 元素判断
is.element(5, x)

## [1] FALSE

is.element(100, x)

## [1] FALSE

is.element(x, y) # length 10

## [1] FALSE FALSE FALSE FALSE FALSE FALSE TRUE FALSE TRUE TRUE

is.element(y, x) # length 8

## [1] FALSE FALSE FALSE TRUE TRUE FALSE FALSE TRUE

集合运算 on 格物堂