根据矩阵元素向igraph中的节点添加属性

时间:2018-08-10 07:31:20

标签: r dataframe matrix igraph

我已经创建了一个网络,希望对节点具有基于我的初始dataframe的属性感兴趣。例如,我的dataframe将每个节点显示为一定数量的关键字,但这是我在创建网络的过程中丢失的信息。在此阶段,我希望将其归因于每个节点,以便在以后的阶段进行进一步分析。

图形示例:

g <- graph.formula(4506-8974, 8974-6345, 7842-4653, 4653-6345, 7842-8974)
V(g)$name <- c("4506", "8974", "6345", "7842", "4653")

另一个困难是我的dataframe采取matrix的形式(尽管从技术上来说还是dataframe)。行是节点,列是关键字。我希望这些列(即关键字)成为每个节点的属性:V(g)$keyword。如果element = 1,则显示为属性;如果element = 0,则不考虑。 有人知道该怎么做吗?我应该使用ifelse函数吗?此外,我认为不要混淆节点ID的顺序很重要,因为否则可能会将关键字属性赋予错误的节点。

df <- data.frame("agriculture" = c(0,1,0,0,0), "arts" = c(0,0,0,1,0), "banks" = c(1,0,1,0,0),
       "cities" = c(0,0,0,1,0), "companies" = c(0,0,0,0,1))
rownames(df) <- c("4506", "8974", "6345", "7842", "4653")

NodeID  agriculture  arts  banks  cities  companies
4506      0          0      1       0        0
8974      1          0      0       0        0
6345      0          0      1       0        0
7842      0          1      0       1        0
4653      0          0      0       0        1

 V(g)$keyword <- keyword based on if the element = 1 for each node

例如,我希望得到的结果是:

V(g)$keyword[1]
[1] "banks"
V(g)$keyword[4]
[4] "arts" "cities"

1 个答案:

答案 0 :(得分:1)

以下代码几乎可以满足您的需求。区别在于V(g)$keyword变成了"list"。因此,其成员的提取与您发布的示例略有不同。

V(g)$keyword <- apply(df, 1, function(x) names(df)[x == 1])

V(g)$keyword[1]
#[[1]]
#[1] "banks

V(g)$keyword[[1]]
#[1] "banks"


V(g)$keyword[4]
#[[1]]
#[1] "arts"   "cities"

V(g)$keyword[[4]]
#[1] "arts"   "cities"