1 Star 4 Fork 11

王布衣 / pandas

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
maximum.go 1.02 KB
一键复制 编辑 原始数据 按行查看 历史
王布衣 提交于 2023-03-08 08:43 . !83#I6L2WO支持ARM框架
package stat
import (
"gitee.com/quant1x/vek"
"gitee.com/quant1x/vek/vek32"
"golang.org/x/exp/slices"
)
// Maximum AVX2版本, 两个序列横向比较最大值
//
// TODO:print(np.maximum(1.4, np.nan)) 输出nan
func Maximum[T Number](f1, f2 []T) []T {
xlen := len(f1)
ylen := len(f2)
// 第找出最大长度
maxLength := xlen
if maxLength < ylen {
maxLength = ylen
}
// 处理默认值
defaultValue := typeDefault[T]()
// 对齐所有长度
if xlen < maxLength {
f1 = Align(f1, T(defaultValue), maxLength)
}
if ylen < maxLength {
f2 = Align(f2, T(defaultValue), maxLength)
}
// 初始化返回值
var s1, s2 any
s1 = f1
s2 = f2
var d any
switch fs1 := s1.(type) {
case []float32:
d = vek32.Maximum(fs1, s2.([]float32))
case []float64:
d = vek.Maximum(fs1, s2.([]float64))
default:
// 目前暂时走不到这里
f1 = slices.Clone(f1)
__maximum(f1, f2)
d = f1
}
return d.([]T)
}
func __maximum[T Number](x, y []T) {
for i := 0; i < len(x); i++ {
if y[i] > x[i] {
x[i] = y[i]
}
}
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Go
1
https://gitee.com/quant1x/pandas.git
git@gitee.com:quant1x/pandas.git
quant1x
pandas
pandas
v0.9.21

搜索帮助

344bd9b3 5694891 D2dac590 5694891