我写的 Python 代码，同事都说好-百木园

原文链接： 我写的 Python 代码，同事都说好

人生苦短，我用 Python。

程序员的追求就是不写代码，早日财务自由。不对，一不小心把实话说出来了，应该是将代码写得简洁，优雅。

Python 程序员的追求则是 Pythonic，正好在 Python 这门语言中，「隐藏」了特别多方法，可以使代码变得简洁，优雅，与众不同。

我在这里总结了一些常用操作，特别是关于列表和字典，分享给大家。

第一个字母大写

这个方法有点意思，无意中发现的。

>>> s = \"programming is awesome\"
>>> print(s.title())
Programming Is Awesome

列表合并

第一种方式：使用 +。

>>> a + b
[1, 2, 3, 4, 5, 6]

第二种方式：使用 extend 关键字。

>>> a.extend(b)
>>> a
[1, 2, 3, 4, 5, 6]

后两种方式明显更加优雅，推荐使用。需要说明的一点是，如果列表很大的话，+ 会比较慢，使用 extend 更好。

列表元素去重

使用 set() 对列表元素进行去重。

>>> a = [1, 2, 3, 4, 2, 3]
>>> list(set(a))
[1, 2, 3, 4]

列表排序

使用 sort() 或内建函数 sorted() 对列表进行排序。它们之间的区别有两点：

sort() 方法是对原列表进行操作，而 sorted() 方法会返回一个新列表，不是在原来的基础上进行操作。
sort() 是应用在列表上的方法，而 sorted() 可以对所有可迭代的对象进行排序操作。

# sort()
>>> a = [1, 2, 3, 4, 2, 3]
>>> a.sort()
>>> a
[1, 2, 2, 3, 3, 4]
>>>
>>> a = [1, 2, 3, 4, 2, 3]
>>> a.sort(reverse=True)
>>> a
[4, 3, 3, 2, 2, 1]

# sorted()
>>> a = [1, 2, 3, 4, 2, 3]
>>> sorted(a)
[1, 2, 2, 3, 3, 4]
>>> a = [1, 2, 3, 4, 2, 3]
>>> sorted(a, reverse=True)
[4, 3, 3, 2, 2, 1]

遍历列表的索引和元素对

使用 enumerate() 函数可以同时输出索引和元素值。

>>> a = [\'python\', \'go\', \'java\']
>>> for i, v in enumerate(a):
...     print(i, v)

# output
0 python
1 go
2 java

查找列表中出现最频繁的元素

使用 max() 函数可以快速查找出一个列表中出现频率最高的某个元素。

>>> a = [1, 2, 3, 4, 3, 4, 5, 4, 4, 2]
>>> b = max(set(a), key=a.count)
>>> b
4

需要说明的一点是，当列表中有两个元素出现的次数相同时，会返回第一个出现的元素。

>>> a = [1, 2]
>>> b = max(set(a), key=a.count)
>>> b
1

统计列表中所有元素的出现次数

前面的代码给出了出现最频繁的值。如果想要知道列表中所有元素的出现次数，那么可以使用 collections 模块。

collections 是 Python 中的一个宝藏模块，它提供了很多特性。Counter 方法正好可以完美解决这个需求。

>>> from collections import Counter
>>>
>>> a = [1, 2, 3, 4, 3, 4, 5, 4, 4, 2]
>>> Counter(a)
Counter({4: 4, 2: 2, 3: 2, 1: 1, 5: 1})

将两个列表合并为字典

使用 zip() 函数，可以将两个列表合并成字典。

>>> a = [\'one\', \'tow\', \'three\']
>>> b = [1, 2, 3]
>>> dict(zip(a, b))
{\'one\': 1, \'tow\': 2, \'three\': 3}

求两个列表的交集，并集和差集

# list_operate.py

def main():
    list_a = [1, 2, 3, 4, 5]
    list_b = [4, 5, 6, 7, 8]

    # 求交集的两种方式
    res_a = [i for i in list_a if i in list_b]
    res_b = list(set(list_a).intersection(set(list_b)))

    print(f\"res_a is: {res_a}\")
    print(f\"res_b is: {res_b}\")

    # 求并集
    res_c = list(set(list_a).union(set(list_b)))
    print(f\"res_c is: {res_c}\")

    # 求差集的两种方式，在B中但不在A中
    res_d = [i for i in list_b if i not in list_a]
    res_e = list(set(list_b).difference(set(list_a)))

    print(f\"res_d is: {res_d}\")
    print(f\"res_e is: {res_e}\")


if __name__ == \'__main__\':
    main()

字典创建

# 1、创建空字典
a = {}
b = dict()

# 2、有初始值，从输入的便利程度来说，我更喜欢第二种
a = {\'a\': 1, \'b\': 2, \'c\': 3}
b = dict(a=1, b=2, c=3)

# 3、key 来自一个列表，而 value 相同, 使用 fromkeys，那是相当的优雅
keys = [\'a\', \'b\', \'c\']
value = 100
d = dict.fromkeys(keys, value)

# 4、key 来自一个列表，而 value 也是一个列表，使用 zip
keys = [\'a\', \'b\', \'c\']
values = [1, 2, 3]
d = dict(zip(keys, values))

字典合并

m = {\'a\': 1}
n = {\'b\': 2, \'c\': 3}

# 合并，两种方式
# 1、使用 update
m.update(n)
# 2、使用 **
{**m, **n}

判断 key 是否存在

在 Python2 中判断某个 key 是否存在，可以使用 has_key，但这个方法在 Python3 中已经被移除了。

另一种方法是使用 in 关键字，不仅兼容 Python2 和 Python3，速度还更快，强烈推荐。

d = {\'a\': 1, \'b\': 2}
if \'a\' in d:
    print(\'hello\')

获取字典中的值

d = {\'a\': 1, \'b\': 2}

# 1、直接用 key 取值，但这种方式不好，如果 key 不存在会报错，推荐使用 get
a = d[\'a\']

# 2、使用 get，如果 key 不存在还可以赋默认值
a = d.get(\'a\')
c = d.get(\'c\', 3)

字典遍历

d = {\'a\': 1, \'b\': 2, \'c\': 3}

# 遍历 key
for key in d.keys():
    pass

# 遍历 value
for value in d.values():
    pass

# 遍历 key 和 value
for key, value in d.items():
    pass

字典推导式

列表推导式和字典推导式是我相当喜欢的功能，简洁高效。map 和 filter 我都已经快不会用了。

l = [1, 2, 3]
{n: n * n for n in l}
{1: 1, 2: 4, 3: 9}

字典按 key 或 value 排序

d = {\'a\': 1, \'b\': 2, \'e\': 9, \'c\': 5, \'d\': 7}

# 按 key 排序
sorted(d.items(), key=lambda t: t[0])
# 按 key 倒序
sorted(d.items(), key=lambda t: t[0], reverse=True)

# 按 value 排序
sorted(d.items(), key=lambda t: t[1])

还有一个需求是我在开发过程经常碰到的，就是有一个列表，列表的元素是字典，然后按字典的 value 对列表进行排序。

l = [{\'name\': \'a\', \'count\': 4}, {\'name\': \'b\', \'count\': 1}, {\'name\': \'d\', \'count\': 2}, {\'name\': \'c\', \'count\': 6}]
sorted(l, key=lambda e: e.__getitem__(\'count\'))
# 倒序
sorted(l, key=lambda e: e.__getitem__(\'count\'), reverse=True)

以上就是本文的全部内容，如果觉得还不错的话，欢迎点赞，转发和关注，感谢支持。

推荐阅读：

Go 学习路线（2022）
Python 学习路线（2022）

来源：https://www.cnblogs.com/alwaysbeta/p/16046919.html
本站部分图文来源于网络，如有侵权请联系删除。

我写的 Python 代码，同事都说好