以下是一个示例代码,其中使用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
列表中的结果。
下一篇:按另一个向量对向量进行排序