不滚动边缘的np.roll
创始人
2024-12-24 18:01:56
0

要实现一个不滚动边缘的np.roll函数,可以使用np.pad函数来对数组进行填充,然后再进行滚动操作。下面是一个示例代码:

import numpy as np

def roll_without_edge(arr, shift):
    # 计算填充的数量
    pad_width = abs(shift)
    
    # 根据滚动方向选择填充位置
    if shift > 0:
        padded_arr = np.pad(arr, pad_width, mode='constant', constant_values=0)
        rolled_arr = np.roll(padded_arr, shift)
        result = rolled_arr[pad_width:len(arr)+pad_width]
    else:
        padded_arr = np.pad(arr, pad_width, mode='constant', constant_values=0)
        rolled_arr = np.roll(padded_arr, shift)
        result = rolled_arr[pad_width:len(arr)+pad_width]

    return result

# 示例用法
arr = np.array([1, 2, 3, 4, 5])
shift = 2
result = roll_without_edge(arr, shift)
print(result)

在上面的代码中,roll_without_edge函数接受一个数组和滚动的位移作为输入,并返回滚动后的结果。首先,我们计算填充的数量(即位移的绝对值),然后根据滚动的方向选择填充位置。我们使用np.pad函数对数组进行填充,然后使用np.roll函数进行滚动操作。最后,根据填充的数量截取出滚动后的结果,并返回。

在示例用法中,我们定义了一个数组arr和一个位移shift,然后调用roll_without_edge函数进行滚动操作,并打印出结果。输出结果为[4, 5, 1, 2, 3],表示数组滚动两个位置后的结果。

相关内容

热门资讯

前端-session、jwt 目录:   (1)session (2&#x...
linux入门---制作进度条 了解缓冲区 我们首先来看看下面的操作: 我们首先创建了一个文件并在这个文件里面添加了...
关于测试,我发现了哪些新大陆 关于测试 平常也只是听说过一些关于测试的术语,但并没有使用过测试工具。偶然看到编程老师...
前缀和与对数器与二分法 1. 前缀和 假设有一个数组,我们想大量频繁的去访问L到R这个区间的和,...
nodejs:本地安装nvm实... 一、背景-使用不同版本node的原因 vue3+ts、nuxt3版本,node...
JAVA集合知识整理 Java集合知识整理 HashMap相关 HashMap的底层数据结构:jdk1.8之...
无刷直流电机介绍及单片机控制实... 无刷直流电机介绍及单片机控制实例前言基本概念优势与劣势使用寿命基本结构使用单片机控制实例电子调速器&...
fwdiary(2) dp2 1.传纸条  AcWing 275. 传纸条 - AcWing 走两条路,走一条最大的...
常用的DOS命令 常用的DOS命令 DOS(Disk Operating System,磁...
<C++> 类和对象(下) 1.const成员函数将const修饰的“成员函数”称之为const成员函数,cons...