英文:
Shorthand index to get a cycle of a numpy array
问题
要在NumPy数组中索引所有元素并在最后包括第一个索引，可以使用以下代码：
import numpy as np
a = np.asarray([2, 4, 6])
# 一种解决方案是
cycle = np.append(a, a[0])
# 另一种解决方案是
cycle = a[[0, 1, 2, 0]]
# 另一种方法是使用切片操作
cycle = a[:]
请注意，第三种方法使用了切片操作，但没有包括索引0，如果要包括第一个索引，可以使用第一种或第二种方法。
英文:
I want to index all elements in a numpy array and also include the first index at the end. So if I have the array [2, 4, 6] , I want to index the array such that the result is [2, 4, 6, 2].
import numpy as np
a = np.asarray([2,4,6])
# One solution is
cycle = np.append(a, a[0])
# Another solution is
cycle = a[[0, 1, 2, 0]]
# Instead of creating a list, can indexing type be combined?
cycle = a[:+0]
答案1
得分: 2
另一个可能的解决方案：
a = np.asarray([2,4,6])
cycle = np.take(a, np.arange(len(a)+1), mode='wrap')
输出：
[2 4 6 2]
英文:
Another possible solution:
a = np.asarray([2,4,6])
cycle = np.take(a, np.arange(len(a)+1), mode='wrap')
Output:
[2 4 6 2]
通过集体智慧和协作来改善编程学习和解决问题的方式。致力于成为全球开发者共同参与的知识库，让每个人都能够通过互相帮助和分享经验来进步。
评论