Python一些可能用的到的函数系列87 向量移动

最新推荐文章于 2024-01-11 20:00:00 发布

原创

最新推荐文章于 2024-01-11 20:00:00 发布 · 734 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言 #后端

这篇博客介绍了如何在Python中处理时序向量，通过设定步长和填充值来实现向量的前后移动，生成矩阵的效果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

说明

假设获得了一份时序向量，我喜欢可以向前和向后移动，形成一个矩阵。

内容

输入一个向量，按照指定的步长和填充值进行移动。

import numpy as np 
# 向量滑动位移为矩阵  direction=1 向过去移动，-1向未来移动
def roll_arr2mat(some_arr, step=1, fillna= 0, direction=1, is_sort=True):
    # 确保向量为一维
    some_arr1 = np.array(some_arr).ravel()
    arr_len = len(some_arr1)
    
    res_list = []
    vec = some_arr1
    # move past
    if direction == 1