#plusone() 関数の改良
どうもどこかで見たはずの，前回の tuple でやる方式は，スタンダードではないらしいので，list に変えて，更に拡張・改良する．
##1 前回の関数
以前の関数．出力はコメントアウトで表現．名前は plusone_tuple() に変えた．
import numpy as np
from sympy.combinatorics import *
from sympy import *
init_printing(pretty_print=False)
n = 3
def plusone_tuple(x):
X = tuple([0])
for i in range(n):
a = tuple([tuple(x)[i] + 1])
X = X + a
return Permutation(tuple(X))
print(plusone_tuple(Permutation(0,1,2)))
#(1 2 3)
少し結合が面倒になったり，添字に注意が必要だったりするが，tuple でやるよりも list でやったほうがいい．
とりあえず，内包表記を使って3次対称群の元を[0, 1, 2]に掛けたものを書いてみる．
A = PermutationGroup(SymmetricGroup(n)[0], SymmetricGroup(n)[1])._elements
print(A)
[list(A[i]) for i in range(len(A))]
'''
[Permutation(2), Permutation(0, 1, 2), Permutation(0, 2, 1), Permutation(1, 2), Permutation(2)(0, 1), Permutation(0, 2)]
[[0, 1, 2], [1, 2, 0], [2, 0, 1], [0, 2, 1], [1, 0, 2], [2, 1, 0]]
'''
##2 今回の関数
list にした関数と複数の置換のリストを扱えるよう改良した関数．
from sympy.combinatorics import *
from sympy import *
init_printing(pretty_print=False)
def plusone(x):
add_one = [(i > 0) * (list(x)[i - 1] + 1) for i in range(1 + len(list(x)))]
#(i > 0) is a step function.
return Permutation(add_one)
n = 3
def plusone_list(x):
return plusone(x)
def plusone_multi(list_of_permutations):
return [plusone(list_of_permutations[i]) for i in range(len(list_of_permutations))]
print(A)
AA = plusone_multi(A)
AA
'''
[Permutation(2), Permutation(0, 1, 2), Permutation(0, 2, 1), Permutation(1, 2), Permutation(2)(0, 1), Permutation(0, 2)]
[(3), (1 2 3), (1 3 2), (2 3), (3)(1 2), (1 3)]
'''
ちなみに面倒くさいがこうしても同じである．
generate_Sym3_perm = PermutationGroup(plusone(SymmetricGroup(n)[0]), plusone(SymmetricGroup(n)[1]))._elements
generate_Sym3_perm
'''
[(3), (1 2 3), (1 3 2), (2 3), (3)(1 2), (1 3)]
'''
以下の list_multi() が複数の置換のリストを list に変換する．これで tuple を介在させずに，list で統一的に対称群の元を扱えるようになった．このほうがどう考えてもシンプルである．
def list_multi(list_of_permutations):
return [list(list_of_permutations[i]) for i in range(len(list_of_permutations))]
print(list_multi(A))
print(list_multi(AA))
'''
[[0, 1, 2], [1, 2, 0], [2, 0, 1], [0, 2, 1], [1, 0, 2], [2, 1, 0]]
[[0, 1, 2, 3], [0, 2, 3, 1], [0, 3, 1, 2], [0, 1, 3, 2], [0, 2, 1, 3], [0, 3, 2, 1]]
'''
##3 3次対称群をシンプルに計算
3次対称群の要素を表示する．
Sym3_perm = SymmetricGroup(n)._elements
Sym3_perm
'''
[(2), (0 1 2), (0 2 1), (1 2), (2)(0 1), (0 2)]
'''
1から始まる表示にして，リスト（[0,1,2,3]に置換を掛けた結果）に直してみる．
Sym3_list = list_multi(plusone_multi(Sym3_perm))
Sym3_list
'''
[[0, 1, 2, 3], [0, 2, 3, 1], [0, 3, 1, 2], [0, 1, 3, 2], [0, 2, 1, 3], [0, 3, 2, 1]]
'''
得たかった結果がえられた．
置換とリストを相互に変換できなくなるので，0は残しておく．プログラムは，「3を固定して1, 2を入れ替え」を先に扱い，「2を固定して1, 3を入れ替え」= (1 3) を後に置いている．元の命名が課題になりそうだ．元の数が24になる4次対称群ならなおさらである．
#考察
np.array で出来ることに気づいたので計算時間を測定して比較してみる．
まずは上の．
%%timeit
Sym3_list = list_multi(plusone_multi(Sym3_perm))
Sym3_list
#129 µs ± 6.92 µs per loop (mean ± std. dev. of 7 runs, 10000 loops each)
#132 µs ± 13.1 µs per loop (mean ± std. dev. of 7 runs, 10000 loops each)
#132 µs ± 13.1 µs per loop (mean ± std. dev. of 7 runs, 10000 loops each)
numpy.
%%timeit
Sym3_list_np = np.array(plusone_multi(Sym3_perm)).tolist()
Sym3_list_np
#139 µs ± 10.7 µs per loop (mean ± std. dev. of 7 runs, 10000 loops each)
#135 µs ± 5.21 µs per loop (mean ± std. dev. of 7 runs, 10000 loops each)
#143 µs ± 8.14 µs per loop (mean ± std. dev. of 7 runs, 10000 loops each)
・1回目は list_multi → np.array
・2回目は np.array → list_multi
・3回目は並列に実行してみた．
誤差の範囲で同じ，と言えそうだ．
ただ，np.array を使って少し重くなるのか，list_multi() を用いたほうがこころなしか速く，安定しているように思う．ここでは，numpy を import する必要もなさそうだ．
前回記事：
Python de 対称群 - Qiita
次の記事：
Python de 対称群 3 - Qiita