2

我需要做一个排列，看起来像一个 1-N 的序列，其中一个 0 填充到位#k。

这种方法有效，但是使用内置函数有什么更简单的吗？

def permshift(n,k):
   return [0 if x == k else x+(x<k) for x in range(n)]

>>> permshift(7,0)
[0, 1, 2, 3, 4, 5, 6]
>>> permshift(7,1)
[1, 0, 2, 3, 4, 5, 6]
>>> permshift(7,2)
[1, 2, 0, 3, 4, 5, 6]
>>> permshift(7,3)
[1, 2, 3, 0, 4, 5, 6]
>>> permshift(7,4)
[1, 2, 3, 4, 0, 5, 6]
>>> permshift(7,5)
[1, 2, 3, 4, 5, 0, 6]
>>> permshift(7,6)
[1, 2, 3, 4, 5, 6, 0]
4

5 回答 5

4

只需创建一个新列表，然后将其插入0您需要的任何位置。

def permshift(n, k):
    lst = range(1, n)
    lst.insert(k, 0)
    return lst
于 2012-06-28T20:50:29.463 回答
2

更多行，但我认为它更简单：

def permshift(n, k):
   x = range(1, n)
   x.insert(k, 0)
   return x

这是一个比最流行的重复答案更短的有趣答案：

def permshift(n, k):
   x = range(1, n)
   return x[:k] + [0] + x[k:]
于 2012-06-28T20:50:54.293 回答
1

呵呵 - 当这么多人想出相同的解决方案时，这很有趣

def permshift(n,k):
     r = range(1,n)
     r.insert(k,0)
     return r

为了完整起见，如果您真的关心效率，这里有一个生成器表达式

def permshift(n,k):
    for i in xrange(1,n):
        yield i
        if i == k:
            yield 0

print list(permshift(7,2))
于 2012-06-28T20:52:07.643 回答
1

已经有了很好的答案，但这里有一个用于踢球和咯咯笑的单行版本：

permshift = lambda n,k: range(1, k+1) + [0] + range(k+1, n)
于 2012-06-28T20:58:38.060 回答
0 
from itertools import chain

def permshift(n,k):
    return chain(xrange(1,k+1), [0], xrange(k+1,n))
于 2012-06-28T21:20:53.457 回答