差し込み処理(format)で空白を含んだ入力をしたい

Question

前提・実現したいこと

実現したいこと：
・コード内のmod="{} ~ {} + {} + {}"に差し込み処理で変数を埋め込みそれらの組み合わせを出力させようとしています。
・現状のやり方では｛｝の数だけ変数を入力する必要がありますが、
空欄も含んだ入力をして、右辺（~より右側）を3変数だけでなく2変数や1変数の組み合わせも出力もさせたいと思っています（例：出力させたい結果）。

条件：
・代入する変数名は、読み込んだcsvファイル内に記載されている必要があります。
・この段階では必要ありませんが、csvファイル内の各変数には1次元の数値配列が与えられています。

アルゴリズムや、参考コードなどわかる方がいらっしゃればアドバイス・ご教示をぜひお願いします。

出力させたい結果

現状の出力
word1 ~ word2 + word3 + word4  
word1 ~ word2 + word3 + word5  
word1 ~ word2 + word4 + word5  
word1 ~ word3 + word4 + word5  
word2 ~ word1 + word3 + word4  
word2 ~ word1 + word3 + word5  
word2 ~ word1 + word4 + word5  
word2 ~ word3 + word4 + word5  
word3 ~ word1 + word2 + word4  
word3 ~ word1 + word2 + word5  
word3 ~ word1 + word4 + word5  
word3 ~ word2 + word4 + word5  
word4 ~ word1 + word2 + word3  
word4 ~ word1 + word2 + word5  
word4 ~ word1 + word3 + word5  
word4 ~ word2 + word3 + word5  
word5 ~ word1 + word2 + word3  
word5 ~ word1 + word2 + word4  
word5 ~ word1 + word3 + word4  
word5 ~ word2 + word3 + word4  

出力させたい結果（例：word1に関してのみ）
word1 ~ word2 + word3 + word4
word1 ~ word2 + word3 + word5  
word1 ~ word2 + word4 + word5
word1 ~ word3 + word4 + word5
word1 ~ word2 + word3
word1 ~ word2 + word4  
word1 ~ word2 + word5
word1 ~ word3 + word5
word1 ~ word4 + word5
word1 ~ word2
word1 ~ word3
word1 ~ word4
word1 ~ word5

現状のコード

import itertools
from scipy.special import perm, comb
import pandas as pd

#データ読み込み
test = pd.read_csv('C:/Users/~~~')

#リスト作成（word1~word5は読み込んだcsvファイル内に記載されている）
word1 = 'word1'
word2 = 'word2'
word3 = 'word3'
word4 = 'word4'
word5 = 'word5'
word_list0 = [word1, word2, word3, word4, word5]

word_list_p0 = list(itertools.permutations(word_list0, 1))
word_list_p1 = list(itertools.combinations(word_list0, 3))

#組み合わせ作成
for i in range(perm(len(word_list0), 1, exact=True)):
    for j in range(comb(len(word_list0), 3, exact=True)):
        y0 = word_list_p0[i]
        y1 = word_list_p1[j]

        mod="{} ~ {} + {} + {}"

        mod=(mod.format(y0[0],y1[0],y1[1],y1[2]))

        if y0[0] != y1[0] and y0[0] != y1[1] and y0[0] != y1[2]:
            print(mod) 

Answer 1

strのformatメソッドを使うと以下のように書けます。

>>> print(word_list0)
['word1', 'word2', 'word3', 'word4', 'word5']
>>> from itertools import permutations
>>>
>>> N = 4
>>> for n in range(N, 1, -1):
...     for seq in permutations(word_list0, n):
...         print(("{} ~ {}" + " + {}" * (n-2)).format(*seq))
...
word1 ~ word2 + word3 + word4
word1 ~ word2 + word3 + word5
word1 ~ word2 + word4 + word3
word1 ~ word2 + word4 + word5
word1 ~ word2 + word5 + word3
word1 ~ word2 + word5 + word4
word1 ~ word3 + word2 + word4
word1 ~ word3 + word2 + word5
word1 ~ word3 + word4 + word2
途中略
word4 ~ word1
word4 ~ word2
word4 ~ word3
word4 ~ word5
word5 ~ word1
word5 ~ word2
word5 ~ word3
word5 ~ word4

Comments

[shu-san]

回答ありがとうございます！
非常に短いコードで解決されていて非常に勉強になりました！
このformatメソッドの使い方については不勉強でした。学ばせていただきます。

Answer 2

「空欄も含んだ入力」というものがよく分かりませんが、欲しい結果は以下のようなものでしょうか。

import itertools
import numpy as np

words = [f'word{i+1}' for i in range(5)]

# 最初に左辺となる一つを取り除いたリストから
# すべての組み合わせのパターンに対応する配列位置を列挙しておく
# この配列位置は、左辺となる一つが何であれ共通で利用できる
comb_idx = []
n_words = len(words)
comb_max = n_words - 2
for cnt in range(comb_max, 0, -1):
    for e in itertools.combinations(range(n_words-1), cnt):
        comb_idx.append(list(e)) # 

# 元のリストから左辺となるwordを順に列挙
for leff_word in words:
    # 左辺となる一つを取り除いたリスト
    # ファンシーインデックスが使えるようにnumpy.arrayにする
    comb_words = np.array([w for w in words if w != leff_word])

    # すべての組み合わせ（の配列位置）
    for e in comb_idx:
        print(f'{leff_word} ~ {" + ".join(comb_words[e])}')
    print('-----')

"""
word1 ~ word2 + word3 + word4
word1 ~ word2 + word3 + word5
word1 ~ word2 + word4 + word5
word1 ~ word3 + word4 + word5
word1 ~ word2 + word3
word1 ~ word2 + word4
word1 ~ word2 + word5
word1 ~ word3 + word4
word1 ~ word3 + word5
word1 ~ word4 + word5
word1 ~ word2
word1 ~ word3
word1 ~ word4
word1 ~ word5
-----
word2 ~ word1 + word3 + word4
：
word5 ~ word4
"""

Comments

[shu-san]

回答ありがとうございます！
ご教授いただいた内容で、まさに出力したかった結果を得ることができました。
左辺と右辺で分けることは考えてはいたのですが、ファンシーインデックスも含めて実装方法が勉強不足でした。これを機に学ばせていただきます。

Answer 3

変数が２個の場合、３個の場合、４個の場合と条件で分けて処理すればいいのでは

あるいは、format使わずに普通に文字列連結で、変数のナカミを見ながらつなげていくとか

Comments

[shu-san]

返信遅れましてすみません。回答ありがとうございます！
確かにそうなんですが、力技以外の解決方法を知りたかった次第です。

[y_waiwai]

力技以外はなさそーなきが。。