Python按索引拆分列表：需求实现及尝试方案问题排查

Got it, let's break down why your current groupby approach isn't giving you the expected result, then walk through the simplest solution for your specific case.

Why Your Current Code Fails #

The lambda you're using with groupby — lambda s: s.partition(',')[0] — is trying to split elements by commas, but none of the strings in your input_list contain commas. That means every element's partition result starts with the full string itself, so groupby treats each element as its own group. That's why you're not getting the two groups you want.

The Correct Approach #

Since your requirement is straightforward: put the first element in its own sublist, and all remaining elements in another sublist, you don't need groupby at all. Python list slicing is perfect here:

input_list = ['book flight from Moscow to Paris for less than 200 euro no more than 3 stops', 'Moscow', 'Paris', '200 euro', '3']
output_list = [input_list[:1], input_list[1:]]

Let's break this down:

• input_list[:1] grabs elements from the start up to (but not including) index 1, which gives you a sublist containing just the first element.
• input_list[1:] grabs all elements starting from index 1 to the end of the list, which gives you the sublist of your destination, price, and stop count values.

Running this code will give you exactly your expected output_list:
[['book flight from Moscow to Paris for less than 200 euro no more than 3 stops'], ['Moscow', 'Paris', '200 euro', '3']]

When to Use groupby #

If you ever have a more complex grouping rule (like elements that share a common prefix or need to be grouped by a computed value), groupby is useful. But for this simple split based on position, slicing is cleaner, faster, and easier to read.

内容的提问来源于stack exchange，提问作者Jignasha Royala