溫馨提示×

如何在R語言中創(chuàng)建虛擬變量

小億
119
2024-05-30 12:16:07
欄目: 云計(jì)算

在R語言中，可以使用函數(shù)factor()或as.factor()將一個(gè)分類變量轉(zhuǎn)換為虛擬變量（也稱為啞變量）。以下是一個(gè)示例代碼：

# 創(chuàng)建一個(gè)包含分類變量的數(shù)據(jù)框
data <- data.frame(category = c("A", "B", "A", "C", "B"))

# 將分類變量轉(zhuǎn)換為虛擬變量
data$dummy <- as.factor(data$category)

# 將虛擬變量轉(zhuǎn)換為虛擬變量
dummy_variables <- model.matrix(~ category, data = data)

# 將虛擬變量合并到原始數(shù)據(jù)框中
data <- cbind(data, dummy_variables)

# 查看轉(zhuǎn)換后的數(shù)據(jù)框
print(data)

這段代碼首先創(chuàng)建了一個(gè)包含分類變量的數(shù)據(jù)框data，然后使用as.factor()函數(shù)將category列轉(zhuǎn)換為虛擬變量，并將結(jié)果存儲在新列dummy中。接著，使用model.matrix()函數(shù)將虛擬變量category轉(zhuǎn)換為虛擬變量矩陣dummy_variables，最后將虛擬變量矩陣合并到原始數(shù)據(jù)框data中。

0 踩