按另一个向量对它们进行分组后，将一个向量的值相加_编程开发

按另一个向量对它们进行分组后，将一个向量的值相加

创始人

2024-11-03 07:31:03

0次

以下是一个示例代码，其中使用Python的numpy库来实现向量的分组和相加：

import numpy as np

# 定义两个向量
vector1 = np.array([1, 2, 3, 4, 5])
vector2 = np.array([2, 4, 6, 8, 10])

# 将vector1按照vector2的值进行分组
unique_values = np.unique(vector2)
grouped_vectors = [vector1[vector2 == value] for value in unique_values]

# 将每个分组中的向量相加
summed_vectors = [np.sum(group) for group in grouped_vectors]

print(summed_vectors)

此代码首先使用numpy库定义了两个向量vector1和vector2。然后，使用np.unique函数获取vector2中的唯一值，并将其保存在unique_values变量中。接下来，使用列表推导式将vector1按照vector2的值进行分组，即将相同值的元素放在同一个分组中，并将这些分组保存在grouped_vectors列表中。

最后，使用列表推导式遍历grouped_vectors列表中的每个分组，并使用np.sum函数对每个分组中的向量进行求和。将求和结果保存在summed_vectors列表中。最后，打印出summed_vectors列表中的结果。

上一篇：按另一个向量的首次出现对向量进行分割

下一篇：按另一个向量对向量进行排序

按另一个向量对它们进行分组后，将一个向量的值相加

相关内容

热门资讯