我可以在 Python 中按第二个参数对列表进行排序吗？

Question

我有一个看起来像这样的外部 .txt 文件：

Year of birth (yyyy); id; First name; Second name;
...
Year of birth (yyyy); id; First name; Second name;

我可以制作一个按bbbb部分对所有元素进行排序的新列表吗？

我尝试通过拆分所有元素然后对其进行排序来做到这一点，但就像那样我丢失了连接到该id其他元素。

Answer 1

有很多方法可以使用内置函数和模块或其他答案中提到的第三方库（如pandas来解决此问题。

使用简单的内置函数，您可以通过将文件对象传递给sorted函数来从文件创建一个排序列表：

with open(filename) as file_object:
    lines = sorted(file_object)

这是有效的，因为sorted接受任何序列或可迭代对象。 文件对象是可迭代的并且一次产生一行。

这将从每行的第一个字符开始按字母顺序排序，然后比较第二个，依此类推。

由于您想按分号之间的第二个字符串排序，您需要提取该字符串并按该字符串排序。 有很多方法可以做到这一点。

sorted函数接受一个key参数。 您应该在此处传递一个函数，该函数将为每一行调用，并返回将该行与其他行进行比较的标准。

例如，我们可以将行拆分为一个单独的字符串列表：

>>> line = '1998;1234;Jim;Smith;'
>>> line.split(';')
['1998', '1234', 'Jim', 'Smith']

然后我们想比较列表中的第二个值（我们从0计数，所以第二个值是[1] ）：

>>> line.split(';')[1]
'1234'

创建一个执行此操作的函数，然后我们可以将其传递给sorted作为key ：

def get_id(line):
    return line.split(';')[1]

with open(filename) as file_object:
    lines = sorted(file_object, key=get_id)

可能虽然您希望对lines进行更多处理，以便为每一行都有一个值列表。 Python 有一个内置的csv模块用于处理分隔值（您使用的是分号）：

import csv
from operator import itemgetter

with open(filename) as file_object:
    reader = csv.reader(file_object, delimiter=';')
    rows = sorted(reader, key=itemgetter(1))

在这种情况下，我们可以使用itemgetter而不必编写get_id ，因为csv模块已经为我们完成了所有拆分，我们只需要一个可以获取行中第二个项目的函数。

Answer 2

这是一个可能的答案。

文本文件中项目的示例字符串：

sample = """1998; 1; Jim; Smith;
2001; 2; Andrea; Johnson;
1995; 3; Tom; Jones;"""

我们将通过拆分字符串、拆分行并附加到我们的主数组来创建一个多维数组（列表列表）。

matrix = list()
for record in sample.split('\n'):
    line = [i.strip() for i in record.split(';') if len(i) > 0]
    matrix .append(line)

我们的矩阵：

[
['1998', '1', 'Jim', 'Smith'],
['2001', '2', 'Andrea', 'Johnson'],
['1995', '3', 'Tom', 'Jones']
]

要排序，您可以使用列表的内置方法并实现一个 lambda 函数来选择您想要排序的列。 注意：由于 Python 中的索引从零开始，因此您必须考虑到这一点。 例如，x[1] 表示第二项，或id列：

排序 1 ：ID 降序

matrix.sort(key=lambda x: x[1], reverse=True)

输出 1：

[
['1995', '3', 'Tom', 'Jones'],
['2001', '2', 'Andrea', 'Johnson'],
['1998', '1', 'Jim', 'Smith']
]

排序 2 ：出生年份降序（从小到大）

matrix.sort(key=lambda x: x[0], reverse=False)

输出 2：

[
['2001', '2', 'Andrea', 'Johnson'],
['1998', '1', 'Jim', 'Smith'],
['1995', '3', 'Tom', 'Jones']
]

Answer 3

list1 = """1999;id2;mike;smith;1996;id3;steve;jones;1998;id1;john;smith"""
list2 = list1.split(';')


import pandas as pd
import numpy as np
columns = ['year','id','first','last']
df = pd.DataFrame(np.reshape(list2,(-1,4)),columns=columns)
###### by='id' if you want to sort by id, or replace with('year','id','first','last') whatever you want to sort by.
df.sort_values(by='id',inplace=True)
list3 = df.values.tolist()
list3
output:
[['1998', 'id1', 'john', 'smith'],
 ['1999', 'id2', 'mike', 'smith'],
 ['1996', 'id3', 'steve', 'jones']]