python-3.x - 根据输入整数查找 Python List 的子集 List

Question

我想要一个基于输入整数值的输入列表的子集列表。

例如:

Input List: [3,7,9,11,12]
Input Value: 2
Output List: [1,7,9,11,12]
# 2 is subtracted from first element of list

Input List: [3,7,9,11,12]
Input Value: 5
Output List: [5,9,11,12]
#5 is subtracted from list in sequence manner, first 3 is subtracted then remaining 2 is subtracted from 7 hence output is [5,9,11,12]

Answer 1

如果允许模块，请使用numpy.cumsum():

import numpy as np
input_list = np.asarray([3, 7, 9, 11, 12])
input_integer = 5
output_list = input_list[input_list.cumsum() > input_integer]
output_list[0] -= input_integer - input_list[input_list.cumsum() <= input_integer].sum()
print(output_list)

输出:

[ 5  9 11 12]

我在那里做了什么:

由于要从开始处减去总和，因此使用累积总和将告诉您从何处裁剪列表。

然后设置第一个元素 = 第一个 elem - (input_integer - 裁剪列表的总和)