从不同的索引位置遍历列表中的元素

Question

这应该很容易，但我根本没有找到解决方案。 这是练习：

从“舒适”、“圆形”、“支撑”、“机械”这 4 个词开始，返回所有可能的 2 个词组合的列表。

示例：["comfortable round", "comfortable support", "comfortable machinery", ...]

我已经开始编写一个遍历每个元素的循环，从 index[0] 的元素开始：

words = ["comfortable, ", 'round, ', 'support, ', 'machinery, ']
index_zero= words[0]


for i in words:
    words = index_zero + i
    words_one = index_one + i 
    print(words)

>>> Output=
comfortable, comfortable,
comfortable, round,
comfortable, support,
comfortable, machinery

问题是当我想从第二个元素（'round'）开始迭代时。我尝试过操作索引（index[0] + 1），但当然，它不会返回任何内容，因为元素是字符串。 我知道需要进行从字符串到索引的转换，但我不确定如何。

我也尝试过定义一个函数，但它会返回 None

word_list = ["comfortable, ", 'round, ', 'support, ', 'machinery, ']
index_change = word_list[0]+ 1

def word_variations(set_of_words):
    for i in set_of_words:
        set_of_words = set_of_words[0] + i


set_of_words = word_variations(word_list)   
print(set_of_words)

Answer 1

我认为这会满足您的需求：

def word_variations(word_list):
  combinations = []
  for first_word in word_list:
    for second_word in word_list:
      if first_word != second_word:
        combinations.append(f'{first_word}, {second_word}')

  return combinations

word_list = ["comfortable", "round", "support", "machinery"]
print(word_variations(word_list))

解释：

您需要在函数末尾包含一个 return 语句才能返回一个值。在我的示例函数word_variations() 中，我首先定义了一个名为combinations 的空列表。这将存储我们计算的每个组合。然后我遍历输入word_list 中的所有单词，创建另一个内部循环以再次遍历所有单词，如果first_word 不等于second_word，则将该组合附加到我的combinations 列表中。所有循环完成后，从函数返回完成的列表。

如果我稍微更改代码以在新行上打印每个结果：

def word_variations(word_list):
  combinations = []
  for first_word in word_list:
    for second_word in word_list:
      if first_word != second_word:
        combinations.append(f'{first_word}, {second_word}')

  return combinations

word_list = ["comfortable", "round", "support", "machinery"]

for combo in word_variations(word_list):
  print(combo)

输出是：

comfortable, round
comfortable, support
comfortable, machinery
round, comfortable
round, support
round, machinery
support, comfortable
support, round
support, machinery
machinery, comfortable
machinery, round
machinery, support

Answer 2

如果您想在这样的 Python 循环中使用索引，您应该使用 enumerate 或遍历列表的长度。以下示例将从第二个元素开始循环。

使用enumerate 同时获取索引和单词的示例：

for i, word in enumerate(set_of_words[1:]):

仅使用索引的示例：

for i in range(1, len(set_of_words)):

注意：上面的set_of_words[1:] 是一个slice，它返回从第二个元素开始的列表。

Answer 3

你也可以像这样使用itertools.permutations()

from itertools import permutations

lst = ['comfortable', 'round', 'support', 'machinery']

for i in list(permutations(lst, 2)):
    print(i)