Question

我有一个很大的整数数组，我需要将数组中每10个整数及其相应索引的最大值打印为一对。

ex. (max_value, index of max_value in array)

我可以在前10个整数中找到最大值和相应的索引，但是我在遍历整个数组时遇到问题。

我尝试过使用：

a = some array of integers

split = [a[i:i+10] for i in xrange(0, len(a), 10)] 

for i in split:
    j = max(i) 
    k = i.index(max(i))
    print (j,k)

这个方法的问题是它将我的数组拆分为10个块，因此max_values是正确的，但索引是不准确的（所有索引都在0-10之间。）我需要找到一种方法，这样做不会将我的数组拆分成块，以便保留原始索引。我确信有一种更简单的循环方式来查找最大值，但我似乎无法弄明白。

Answer 1

您需要计算当前窗口之前显示的元素数量。这将完成这项工作：

a=list(range(5,35))
split = [a[i:i+10] for i in xrange(0, len(a), 10)] 

for ind,i in enumerate(split):
    j = max(i) 
    k = i.index(j)
    print (j,k+ind*10)

打印

(14, 9)
(24, 19)
(34, 29)

Answer 2

对当前代码的一个小修改：

a = some array of integers

split = [a[i:i+10] for i in xrange(0, len(a), 10)] 

for index, i in enumerate(split):
    j = max(i) 
    k = i.index(max(i))
    print (j, k+10*index)

Answer 3

因此，通过调试示例数组，我们发现split返回一个像这样的二维列表：

[[1, 2, 3, 4, 5, 6, 7, 8, 9, 10], [11, 12, 13, 14, 15, 16, 17, 18, 19, 20]]

每次for循环运行时，它都按顺序执行其中一个列表。首先它通过第一个内部列表然后第二个内容等等。所以每次for循环跳转到下一个列表时，我们只需添加10.由于列表中可以有超过2个列表，我们存储数字我们需要添加一个变量并在每个循环中添加10：

a = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20]
split = [a[i:i+10] for i in xrange(0, len(a), 10)] 
counter = 0

for i in split:
    j = max(i) 
    k = i.index(max(i))
    print (j,k+counter)
    counter += 10

你可以test it here

Answer 4

您需要循环才能遍历列表，但我们可以更改您的split循环，使其更符合您的需求。

a = some array of integers

split = [a[i:i+10] for i in xrange(0, len(a), 10)] 

for i in range(len(split)):
    #Now instead of being the list, i is the index, so we can use 10*i as a counter
    j = max(split[i]) 
    #j = max(i) 
    k = split[i].index(j) + 10*i #replaced max(i) with j since we already calculated it.
    #k = i.index(max(i))
    print (j,k)

虽然在将来，请为split列表创建一个新名称，因为split已经是python中的一个函数。可能是split_list或separated或其他一些看起来不像split()函数的名称。

Answer 5

toolz包有一个partition_all函数，可以将序列划分为相等大小的元组，因此您可以执行类似的操作。

import toolz
ns = list(range(25))
[max(sublist) for sublist in toolz.partition_all(10, ns)]

这将返回[9, 19, 24]。

Answer 6

任意输入的numpy解决方案：

import numpy as np

a = np.random.randint(1,21,40)  #40 random numbers from 1 to 20

b = a.reshape([4,10])  #shape into chunks 10 numbers long

i = b.argsort()[:,-1]  #take the index of the largest number (last number from argsort) 
                       #  from each chunk. (these don't take into account the reshape)

i += np.arange(0,40,10)  #add back in index offsets due to reshape

out = zip(i, a[i])  #zip together indices and values

Answer 7

您可以通过枚举一次并使用zip将列表分组来简化此操作：

n=10
for grp in zip(*[iter(enumerate(some_list))]*n):
    grp_max_ind, grp_mv=max(grp, key=lambda t: t[1])
    k=[t[1] for t in grp].index(grp_mv)
    print grp_mv, (grp_max_ind, k)

如果您需要生成器（或使用Python 3），请在Python 2中使用izip

from itertools import izip 
for grp in izip(*[iter(enumerate(some_list))]*n):
    grp_max_ind, grp_mv=max(grp, key=lambda t: t[1])
    k=[t[1] for t in grp].index(grp_mv)
    print grp_mv, (grp_max_ind, k)

如果长度不是n

，

Zip将截断最后一个组

Answer 8

使用numpy的示例。首先让我们生成一些数据，即从1到V和长度（值的数量）L的整数：

import numpy as np
V = 1000
L = 45 # method works with arrays not multiples of 10
a = np.random.randint(1, V, size=L)

现在解决大小为N的子数组的问题：

import numpy as np
N = 10 # example "split" size
sa = np.array_split(a, range(N, len(a), N))
sind = [np.argpartition(i, -1)[-1] for i in sa]
ind = [np.ravel_multi_index(i, (len(sa), N)) for i in enumerate(sind)]
vals = np.asarray(a)[np.asarray(ind)]
split_imax = zip(vals, ind) # <-- output

PYTHON - 查找数组中每10个整数的最大值

8 个答案: