Создание матриц

Часть раздела Логика и потоки выполнения путешествия по R на Coddy. Урок 17 из 64.

Матрица — это вектор, расположенный в строках и столбцах, причём каждое значение имеет один и тот же тип. По умолчанию matrix(data, nrow = ...) заполняет её по столбцам:

m <- matrix(1:6, nrow = 2)
print(m)

Результат:

     [,1] [,2] [,3]
[1,]    1    3    5
[2,]    2    4    6

Добавьте byrow = TRUE, чтобы заполнять матрицу построчно, как при записи данных по одной строке в каждой строке:

m <- matrix(1:6, nrow = 2, byrow = TRUE)
print(m)

Результат:

     [,1] [,2] [,3]
[1,]    1    2    3
[2,]    4    5    6

Укажите либо nrow, либо ncol; R вычислит другой параметр по длине данных.

dim(m) возвращает количество строк и столбцов, а nrow() и ncol() возвращают их по отдельности. matrix(0, nrow = 2, ncol = 3) повторяет одно значение, чтобы заполнить матрицу:

m <- matrix(0, nrow = 2, ncol = 3)
print(dim(m))
print(nrow(m) * ncol(m))
print(length(m))

Результат:

[1] 2 3
[1] 6
[1] 6

rbind() объединяет векторы в строки, а cbind() — в столбцы, что позволяет создать матрицу из отдельных векторов:

a <- c(1, 2, 3)
b <- c(4, 5, 6)
print(rbind(a, b))
print(dim(cbind(a, b)))

Результат:

  [,1] [,2] [,3]
a    1    2    3
b    4    5    6
[1] 3 2

Задание

Легко

Реализуйте make_grid(values, n_cols). Верните матрицу с n_cols столбцами, в которой values расположены построчно. Количество значений всегда кратно n_cols.

Предоставленный код считывает значения из строки, разделённой запятыми, а n_cols — из следующей строки. Он выводит каждую строку возвращённой матрицы в отдельной строке, а затем её размер в формате rows x columns.

Попробуйте сами

make_grid <- function(values, n_cols) {
  # Напишите свой код здесь
  matrix(values)
}

# Предоставленный код ввода/вывода: оставьте его как есть
input <- suppressWarnings(readLines(file("stdin")))
values <- as.numeric(strsplit(input[1], ",")[[1]])
m <- make_grid(values, as.numeric(input[2]))
for (i in seq_len(nrow(m))) cat(m[i, ], "\n")
cat(nrow(m), "x", ncol(m), "\n")