Adding Thousand Separators using itertools

この問題が面白そうだなと思い、私もPythonで挑戦してみました。目的は、数字にコンマを振ることです。

「Python のジェネレータ (2)」に引き続き、ジェネレータに慣れるための練習。o(+_+)o どかにまた例題はないかと散策していたら、「数字にコンマを振る」が良さげ。
Python のジェネレータ (3) - 数字にコンマを振る | すぐに忘れる脳みそのためのメモ

当初は、先日Schemeでコーディングしたこの問題と同様、継続渡し形式でコードを書く良い問題であるのではないかと思っていました。しかし、これが大変難しい。そのため、イテレータ、itertoolsモジュールを積極的に使う方針としてみました。出来上がったコードは次のようなものです。Python3.0を使っています。

from itertools import chain, cycle, islice

def thousand_separated(n):
    return ''.join(chain(*zip(reversed(str(n)),
                              cycle(('', '', ',')))))[::-1].lstrip(',')

うーん、いいのか悪いのか。最後にstr.lstrip()で整形しているのが悲しいところ。

テストコードを含めたコードとその結果を以下に。数値の生成にもitertoolsで。

thousand_separated.py:

#!/usr/local/bin/python3.0 -t

from itertools import chain, cycle, islice

def thousand_separated(n):
    return ''.join(chain(*zip(reversed(str(n)),
                              cycle(('', '', ',')))))[::-1].lstrip(',')

if __name__ == '__main__':
    for i in range(20):
        n = int(''.join(islice(cycle('1234567890'),
                               i+1)))
        print('%27s' % thousand_separated(n))

> ./thousand_separated.py
                          1
                         12
                        123
                      1,234
                     12,345
                    123,456
                  1,234,567
                 12,345,678
                123,456,789
              1,234,567,890
             12,345,678,901
            123,456,789,012
          1,234,567,890,123
         12,345,678,901,234
        123,456,789,012,345
      1,234,567,890,123,456
     12,345,678,901,234,567
    123,456,789,012,345,678
  1,234,567,890,123,456,789
 12,345,678,901,234,567,890

さて、関数内で何が起こっているかをPostScriptで可視してみました。目的は数値にコンマを挿入することでした。