python – 如何同时获取元组的第一个和最后一个元素

我需要获得任意大小的numpy.ndarray的第一个和最后一个维度.

如果我有形状(A)=(3,4,4,4,4,4,4,3)
我的第一个想法是做结果=形状(A)[0,-1],但这似乎不适用于元组,为什么不呢?

有没有比这更简洁的方式

s=shape(A)
result=(s[0], s[-1])

谢谢你的帮助

最佳答案
我不知道什么是错的

(s[0], s[-1])

另一种选择是使用operator.itemgetter():

from operator import itemgetter
itemgetter(0, -1)(s)

不过,我认为这不会更好. (如果不计算实例化itemgetter实例所需的时间,可能会稍快一些,如果经常需要这个操作,可以重复使用.)

转载注明原文:python – 如何同时获取元组的第一个和最后一个元素 - 代码日志