2

整数のいくつかのタプルといくつかの整数で構成されるタプルがあります.今、((1, 2), 3, (4, 5), 6)そこからすべての整数が必要です。私が書いた:

def get_all_items(iterable):
    t = []

    for each in iterable:
        if type(each) == int:
            t.append(each)
        else:
            t.extend(each)

    return tuple(t)

わたしにはできる。これを行うより良い方法はありますか?

4

5 に答える 5

3

チートの方法を忘れないでください

>>> from ast import literal_eval
>>> t = ((1, 2), 3, (4, 5), 6)
>>> literal_eval(repr(t).translate(None, '()'))
(1, 2, 3, 4, 5, 6)
于 2013-10-15T04:05:50.873 に答える
1

I think your way is fine. Here's another way, using recursion, which will work for an arbitrarily-deeply-nested iterable structure:

def get_all_items(iterable):
    try:
        result = []
        for element in iterable:
            result += detuple(element)
        return result
    except TypeError:
        return [iterable]

Also, it may be useful to know that the operation you're describing is known as "flattening" a data structure.

于 2013-10-15T03:19:05.323 に答える
1

このようにコードを単純化できます

def get_all_items(iterable):
    t = []
    for each in iterable:
        t.extend(list(each) if isinstance(each, tuple) else [each])
    return tuple(t)

print get_all_items(((1, 2), 3, (4, 5), 6))

出力

(1, 2, 3, 4, 5, 6)
于 2013-10-15T03:49:09.707 に答える
1
from itertools import chain

def flatten(items):
  def renest():
    for item in items:
      try:
        yield iter(item)
      except TypeError:
        yield iter([item])
  return list(chain.from_iterable(renest()))
于 2013-10-15T03:12:04.563 に答える
1
import itertools
itertools.chain.from_iterable(
    item if hasattr(item, '__iter__') else (item,)
        for item in iterable 
)
于 2013-10-15T03:40:50.100 に答える