ass_series
是Gempy的Stack
类中的一个方法,用于将多个数据系列按照指定的权重系数相加。
ass_series(data: numpy.ndarray, weights: numpy.ndarray) -> numpy.ndarray
data
:一个numpy的n维数组(n至少为2),表示多个数据系列。其中,最后一维表示具体的数据,其他维度表示数据的不同特征。weights
:一个与data
的最后一维长度相等的numpy数组,表示各个数据系列的权重系数。data
具有相同的维度及长度。其中,最后一维表示加权后的结果。import numpy as np
from gempy.core.data_modules.stack import Stack
stack = Stack()
data1 = np.random.rand(3, 4, 2)
data2 = np.random.rand(3, 4, 2)
weights = np.array([0.6, 0.4])
result = stack.ass_series(np.stack([data1, data2]), weights)
上述示例中,我们生成了两个shape分别为(3, 4, 2)
的numpy数组data1
和data2
,以及一个权重系数数组weights
。我们将data1
和data2
在第0维压缩起来,形成一个shape为(2, 3, 4, 2)
的numpy数组,将其作为参数传给ass_series
方法。输出的result
将是一个shape为(3, 4, 2)
的numpy数组,表示两个数据系列的按0.6:0.4加权的结果。