以下是一个R函数的示例,用于检查一个矩阵是否只包含0和1:
check_matrix <- function(matrix) {
# 遍历矩阵的每个元素
for (i in 1:nrow(matrix)) {
for (j in 1:ncol(matrix)) {
# 如果元素不是0或1,则返回FALSE
if (matrix[i, j] != 0 && matrix[i, j] != 1) {
return(FALSE)
}
}
}
# 如果所有元素都是0或1,则返回TRUE
return(TRUE)
}
使用示例:
# 创建一个只包含0和1的矩阵
matrix1 <- matrix(c(0, 1, 1, 0), nrow = 2, ncol = 2)
print(check_matrix(matrix1)) # 输出TRUE
# 创建一个包含非0和1的矩阵
matrix2 <- matrix(c(0, 1, 2, 0), nrow = 2, ncol = 2)
print(check_matrix(matrix2)) # 输出FALSE
在上述示例中,函数check_matrix
接受一个矩阵作为输入,并使用两个嵌套的for循环遍历矩阵的所有元素。如果任何元素不是0或1,则函数返回FALSE。如果所有元素都是0或1,则函数返回TRUE。
上一篇:编写一个R函数来返回列名称的组。
下一篇:编写一个R函数来修改数据框中的值