在R語言中,可以使用函數factor()
或as.factor()
將一個分類變量轉換為虛擬變量(也稱為啞變量)。以下是一個示例代碼:
# 創建一個包含分類變量的數據框
data <- data.frame(category = c("A", "B", "A", "C", "B"))
# 將分類變量轉換為虛擬變量
data$dummy <- as.factor(data$category)
# 將虛擬變量轉換為虛擬變量
dummy_variables <- model.matrix(~ category, data = data)
# 將虛擬變量合并到原始數據框中
data <- cbind(data, dummy_variables)
# 查看轉換后的數據框
print(data)
這段代碼首先創建了一個包含分類變量的數據框data
,然后使用as.factor()
函數將category
列轉換為虛擬變量,并將結果存儲在新列dummy
中。接著,使用model.matrix()
函數將虛擬變量category
轉換為虛擬變量矩陣dummy_variables
,最后將虛擬變量矩陣合并到原始數據框data
中。