我有array2D = [[1,2,3],[4,5,6]]
。我想要的是一个函数,该函数需要一个索引并返回一维数组中的元素。
Example: fn(0) -> returns [1,4]
fn{1) -> returns [2,5]
我需要一种快速的方法。
答案 0 :(得分:3)
您可以使用lambda
和list comprehension
:
array2D = [[1,2,3],[4,5,6]]
fn = lambda x: [item[x] for item in array2D]
print(fn(0)) # [1, 4]
print(fn(1)) # [2, 5]
print(fn(2)) # [3, 6]
如注释中所建议,您可以将 same 概念与函数定义一起应用:
def fn(x): return [item[x] for item in array2D]
print(fn(0)) # [1, 4]
print(fn(1)) # [2, 5]
print(fn(2)) # [3, 6]
Lambda函数是pretty useful,可让您以非常清晰的方式定义操作。
在我们的示例中,lambda接受变量x
,该变量表示我们想要的item
array2D
的索引
然后,您拥有list comprehension,类似于lambda函数,它们是一个非常强大的工具,并且是python中的 must ()
在这种情况下,您应该预先熟悉PEP-8建议的函数定义。
答案 1 :(得分:1)
以下列表理解将起作用:
def fn(i, lst):
return [sublst[i] for sublst in lst]
>>> array2D = [[1, 2, 3], [4, 5, 6]]
>>> fn(0, array2D)
[1, 4]
>>> fn(1, array2D)
[2, 5]
答案 2 :(得分:1)
您可以使用operator.itemgetter
:
array2D = [[1,2,3],[4,5,6]]
from operator import itemgetter
def fn(x, k):
return list(map(itemgetter(k), x))
fn(array2D, 0) # [1, 4]
如果要定义用于检索特定索引的新功能,可以通过functools.partial
进行操作:
from functools import partial
def fn(x, k):
return list(map(itemgetter(k), x))
get_zero_index = partial(fn, k=0)
get_zero_index(array2D) # [1, 4]
答案 3 :(得分:0)
发电机怎么样?
我们可以使用zip
打包它们,然后创建一个空列表来存储生成的数据:
class myZip(object):
__slots__ = ('zipData', 'interList')
def __init__(self, *args):
self.zipData = zip(*args)
self.interList = []
def __call__(self, index):
try:
return self.interList[index]
except IndexError:
try:
if index == 0:
self.interList.append(next(self.zipData))
return self.interList[index]
for i in range(index-(len(self.interList)-1)):
self.interList.append(next(self.zipData))
return self.interList[index]
except StopIteration:
raise IndexError("index out of range")
def __iter__(self):
for i in self.interList:
yield i
for i in self.zipData:
yield i
array2D = [[1,2,3],[4,5,6]]
a = myZip(*array2D)
print(a(2))
print(a(1))
print(a(0))
---
(3, 6)
(2, 5)
(1, 4)
这样做的好处是我们不需要一次生成所有数据。
答案 4 :(得分:0)
这是我使用切片的2美分(我必须为此使用额外的np.array()
,因为您的原始数据是list
):
array2D = np.array([[1,2,3],[4,5,6]])
def fn(n): return (list(array2D[:,n]))
print (fn(0), fn(1), fn(2))