r - 提取列标题

Question

我有以下数据框(数据)，我想从中提取和打印列标题，其中有非零元素:

  M1 M2 M3 M4
2  0  1  0 1
5  1 -1  0 0
7  0  1  1 0

所需输出:

2: M2 M4
5: M1 M2
7: M2 M3 

到目前为止，我的 R 代码不起作用:

colnames(data)[which(data[2] !=0),]

帮助将不胜感激。谢谢

Answer 1

这将始终返回一个列表:

Map(`[`, list(names(df)), split(col(df)[df != 0],
                                row(df)[df != 0]))

# [[1]]
# [1] "M2" "M4"
# 
# [[2]]
# [1] "M1" "M2"
# 
# [[3]]
# [1] "M2" "M3"

(如果您希望列表共享 setNames(..., rownames(df)) 的行名称，则可以在 df 内换行。)