map及びイテレータについて

Question

drop関数を以下のようなものを返す関数と定義したいです。
drop(n,xs) #ここでは仮にn=2,xs='abcd'とします。
返すもの：n番目以下のxsの要素 #ここでは'cd'

自分が書いたコード
def drop(n, xs):
k = len(xs)
return map(lambda x, i: x, xs, range(n,k))

結果
print(*drop(2, 'abcd'))== a b #本来c dと出力されるのが好ましいです。

この問題を解決するにはどうすればよいでしょうか。

Answer 1

def drop(n, xs):
    return xs[n:]

xsがイテレータになることもあるのなら、itertools.isliceの利用を検討します。

コメントを受けて

第1 引数で与えられた要素数まで、第2 引数に与えられたコレクション/イテレータを走査するイ

テレータを返す関数take()を定義せよ。
take()の結果のイテレータが走査しない残りの部分をイテレータで返すdrop()を定義せよ。

なるほど。第二引数はイテレータを取る場合があるのですね。
とりあえず、ざっくりイテレータの扱い方を紹介します。

>>> lst = [1, 2, 3]
>>> it = iter(lst)
>>> next(it)
1
>>> next(it)
2
>>> list(it)
[3]
>>>

iterを使えば、任意のコレクションからイテレータを取得することができます。
イテレータは状態を記憶するので、一度取り出した箇所の値を再度送出することはありません。

依って、drop関数は次のように簡単に実装できます。
0. 第二引数xsのイテレータitを取る
0. 第一引数n回分だけitをnextに放り込む
0. itを返す

これだけです。
ただし、nがxsの要素数より多い場合はStopIterationを捌いてやる必要があります。

スマートに解きたいなら前述のスライスやitertoolsを活用してみても良いでしょう。

Comments

[退会済みユーザー]

ご回答ありがとうございます
it = drop(0, '')
print(iter(it) == it)
の結果をTrueと表示したいのですが何か対策ございますでしょうか。

[LouiS0616]

イテレータを取得して返せば良いです。

[LouiS0616]

どこかの練習問題なのでしょうか？
それならば、出典を書いて下さい。

[退会済みユーザー]

ご回答ありがとうございます。
大学の課題なので一般に公開されているものではありませんが、簡潔にまとめると課題は以下のように構成されています。
第1 引数で与えられた要素数まで、第2 引数に与えられたコレクション/イテレータを走査するイ
テレータを返す関数take()を定義せよ。
take()の結果のイテレータが走査しない残りの部分をイテレータで返すdrop()を定義せよ。
第1引数に正の整数????、第2引数に文字列sを取り、sの????-gramを生成するイテレータを返す関数ngrams()を、スライスを使わずに、map()とdrop()を用いて実装せよ。
drop()のテストケースとして
it = drop(0, '')
print(iter(it) == it) # True
が与えられていたということです。

[LouiS0616]

回答を修正しました。ご確認ください。

[退会済みユーザー]

ご回答ありがとうございました。
ご回答を参考に自分でも取り組んでみようと思います。