年薪50W+的Python程序员如何写代码

为什么要用Python写代码

没有对比就没有伤害

很多互联网和移动互联网企业对开发效率的要求高于对执行效率的要求。

例子1：hello, world

C的版本：

  1. #include <stdio.h>
  3. int main() {
  4.     printf("hello, world\n");
  5.     return 0;
  6. }

Java的版本：

  1. class Example01 {
  3.     public static void main(String[] args) {
  4.         System.out.println("hello, world");
  5.     }
  6. }

Python的版本：

  1. print('hello, world')
例子2：1-100求和

C的版本：

  1. #include <stdio.h>
  3. int main() {
  4.     int total = 0;
  5.     for (int i = 1; i <= 100; ++i) {
  6.         total += i;
  7.     }
  8.     printf("%d\n", total);
  9.     return 0;
  10. }

Python的版本：

  1. print(sum(range(1, 101)))
例子3：创建和初始化数组（列表）

Java的版本：

  1. import java.util.Arrays;
  3. public class Example03 {
  5.     public static void main(String[] args) {
  6.         boolean[] values = new boolean[10];
  7.         Arrays.fill(values, true);
  8.         System.out.println(Arrays.toString(values));
  10.         int[] numbers = new int[10];
  11.         for (int i = 0; i < numbers.length; ++i) {
  12.             numbers[i] = i + 1;
  13.         }
  14.         System.out.println(Arrays.toString(numbers));
  15.     }
  16. }

Python的版本：

  1. values = [True] * 10
  2. print(values)
  3. numbers = [x for x in range(1, 11)]
  4. print(numbers)
例子4：双色球随机选号

Java的版本：

  1. import java.util.List;
  2. import java.util.ArrayList;
  3. import java.util.Collections;
  4. import java.util.Scanner;
  6. class Example03 {
  8.     /**
  9.      * 产生[min, max)范围的随机整数
  10.      */
  11.     public static int randomInt(int min, int max) {
  12.         return (int) (Math.random() * (max - min) + min);
  13.     }
  15.     /**
  16.      * 输出一组双色球号码
  17.      */
  18.     public static void display(List<Integer> balls) {
  19.         for (int i = 0; i < balls.size(); ++i) {
  20.             System.out.printf("%02d ", balls.get(i));
  21.             if (i == balls.size() - 2) {
  22.                 System.out.print("| ");
  23.             }
  24.         }
  25.         System.out.println();
  26.     }
  28.     /**
  29.      * 生成一组随机号码
  30.      */
  31.     public static List<Integer> generate() {
  32.         List<Integer> redBalls = new ArrayList<>();
  33.         for (int i = 1; i <= 33; ++i) {
  34.             redBalls.add(i);
  35.         }
  36.         List<Integer> selectedBalls = new ArrayList<>();
  37.         for (int i = 0; i < 6; ++i) {
  38.             selectedBalls.add(redBalls.remove(randomInt(0, redBalls.size())));
  39.         }
  40.         Collections.sort(selectedBalls);
  41.         selectedBalls.add(randomInt(1, 17));
  42.         return selectedBalls;
  43.     }
  45.     public static void main(String[] args) {
  46.         try (Scanner sc = new Scanner(System.in)) {
  47.             System.out.print("机选几注: ");
  48.             int num = sc.nextInt();
  49.             for (int i = 0; i < num; ++i) {
  50.                 display(generate());
  51.             }
  52.         }
  53.     }
  54. }

Python的版本：

  1. from random import randint, sample
  4. def generate():
  5.     """生成一组随机号码"""
  6.     red_balls = [x for x in range(1, 34)]
  7.     selected_balls = sample(red_balls, 6)
  8.     selected_balls.sort()
  9.     selected_balls.append(randint(1, 16))
  10.     return selected_balls
  13. def display(balls):
  14.     """输出一组双色球号码"""
  15.     for index, ball in enumerate(balls):
  16.         print(f'{ball:0>2d}', end=' ')
  17.         if index == len(balls) - 2:
  18.             print('|', end=' ')
  19.     print()
  22. num = int(input('机选几注: '))
  23. for _ in range(num):
  24.     display(generate())

温馨提示：珍爱生命，远离任何形式的赌博。

例子5：实现一个简单的HTTP服务器。

Java的版本：

说明：JDK 1.6以前，需要通过套接字编程来实现，具体又可以分为多线程和NIO两种做法。JDK 1.6以后，可以使用com.sun.net.httpserver包提供的HttpServer类来实现。

  1. import com.sun.net.httpserver.HttpExchange;
  2. import com.sun.net.httpserver.HttpHandler;
  3. import com.sun.net.httpserver.HttpServer;
  5. import java.io.IOException;
  6. import java.io.OutputStream;
  7. import java.net.InetSocketAddress;
  9. class Example05 {
  11.     public static void main(String[] arg) throws Exception {
  12.         HttpServer server = HttpServer.create(new InetSocketAddress(8000), 0);
  13.         server.createContext("/", new RequestHandler());
  14.         server.start();
  15.     }
  17.     static class RequestHandler implements HttpHandler {
  19.         @Override
  20.         public void handle(HttpExchange exchange) throws IOException {
  21.             String response = "<h1>hello, world</h1>";
  22.             exchange.sendResponseHeaders(200, 0);
  23.             try (OutputStream os = exchange.getResponseBody()) {
  24.                 os.write(response.getBytes());
  25.             }
  26.         }
  27.     }
  28. }

Python的版本：

  1. from http.server import HTTPServer, SimpleHTTPRequestHandler
  4. class RequestHandler(SimpleHTTPRequestHandler):
  6.     def do_GET(self):
  7.         self.send_response(200)
  8.         self.end_headers()
  9.         self.wfile.write('<h1>hello, world</h1>'.encode())
  12. server = HTTPServer(('', 8000), RequestHandler)
  13. server.serve_forever()

或

  1. python3 -m http.server 8000

一行Python代码可以做什么

很多时候，你的问题只需一行Python代码就能解决。

  1. # 一行代码实现求阶乘函数
  2. fac = lambda x: __import__('functools').reduce(int.__mul__, range(1, x + 1), 1)
  4. # 一行代码实现求最大公约数函数
  5. gcd = lambda x, y: y % x and gcd(y % x, x) or x
  7. # 一行代码实现判断素数的函数
  8. is_prime = lambda x: x > 1 and not [f for f in range(2, int(x ** 0.5) + 1) if x % f == 0]
  10. # 一行代码实现快速排序
  11. quick_sort = lambda items: len(items) and quick_sort([x for x in items[1:] if x < items[0]]) + [items[0]] + quick_sort([x for x in items[1:] if x > items[0]]) or items
  13. # 生成FizzBuzz列表
  14. ['Fizz'[x % 3 * 4:] + 'Buzz'[x % 5 * 4:] or x for x in range(1, 101)]

设计模式从未如此简单

Python是动态类型语言，大量的设计模式在Python中被简化或弱化。

思考：如何优化下面的代码。

  1. def fib(num):
  2.     if num in (1, 2):
  3.         return 1
  4.     return fib(num - 1) + fib(num - 2)

代理模式在Python中可以通过内置的或自定义的装饰器来实现。

  1. from functools import lru_cache
  4. @lru_cache()
  5. def fib(num):
  6.     if num in (1, 2):
  7.         return 1
  8.     return fib(num - 1) + fib(num - 2)
  11. for n in range(1, 121):
  12.     print(f'{n}: {fib(n)}')

说明：通过Python标准库functools模块的lru_cache装饰器为fib函数加上缓存代理，缓存函数执行的中间结果，优化代码的性能。

单例模式在Python中可以通过自定义的装饰器或元类来实现。

  1. from functools import wraps
  2. from threading import RLock
  5. def singleton(cls):
  6.     instances = {}
  7.     lock = RLock()
  9.     @wraps(cls)
  10.     def wrapper(*args, **kwargs):
  11.         if cls not in instances:
  12.             with lock:
  13.                 if cls not in instances:
  14.                     instances[cls] = cls(*args, **kwargs)
  15.         return instances[cls]

说明：需要实现单例模式的类只需要添加上面的装饰器即可。

原型模式在Python中可以通过元类来实现。

  1. import copy
  4. class PrototypeMeta(type):
  6.     def __init__(cls, *args, **kwargs):
  7.         super().__init__(*args, **kwargs)
  8.         cls.clone = lambda self, is_deep=True: \
  9.             copy.deepcopy(self) if is_deep else copy.copy(self)

说明：通过元类给指定了metaclass=PrototypeMeta的类添加一个clone方法实现对象克隆，利用Python标准库copy模块的copy和deepcopy分别实现浅拷贝和深拷贝。

数据采集和数据分析从未如此简单

网络数据采集是Python最擅长的领域之一。

例子：获取豆瓣电影“Top250”。

  1. import random
  2. import time
  4. import requests
  5. from bs4 import BeautifulSoup
  7. for page in range(10):
  8.     resp = requests.get(
  9.         url=f'https://movie.douban.com/top250?start={25 * page}',
  10.         headers={'User-Agent': 'BaiduSpider'}
  11.     )
  12.     soup = BeautifulSoup(resp.text, "lxml")
  13.     for elem in soup.select('a > span.title:nth-child(1)'):
  14.         print(elem.text)
  15.     time.sleep(random.random() * 5)

利用NumPy、Pandas、Matplotlib可以轻松实现数据分析和可视化。

年薪50W+的Python程序员如何写代码 - 图1

写出Python代码的正确姿势

用Python写代码就要写出Pythonic的代码。

姿势1：选择结构的正确姿势

跨界开发者的代码：

  1. name = 'jackfrued'
  2. fruits = ['apple', 'orange', 'grape']
  3. owners = {'name': '骆昊', 'age': 40, 'gender': True}
  4. if name != '' and len(fruits) > 0 and len(owners.keys()) > 0:
  5.     print('Jackfrued love fruits.')

Pythonic的代码：

  1. name = 'jackfrued'
  2. fruits = ['apple', 'orange', 'grape']
  3. owners = {'name': '骆昊', 'age': 40, 'gender': True}
  4. if name and fruits and owners:
  5.     print('Jackfrued love fruits.')

姿势2：交换两个变量的正确姿势

跨界开发者的代码：

  1. temp = a
  2. a = b
  3. b = temp

或

  1. a = a ^ b
  2. b = a ^ b
  3. a = a ^ b

Pythonic的代码：

  1. a, b = b, a

姿势3：用序列组装字符串的正确姿势

跨界开发者的代码：

  1. chars = ['j', 'a', 'c', 'k', 'f', 'r', 'u', 'e', 'd']
  2. name = ''
  3. for char in chars:
  4.     name += char

Pythonic的代码：

  1. chars = ['j', 'a', 'c', 'k', 'f', 'r', 'u', 'e', 'd']
  2. name = ''.join(chars)

姿势4：遍历列表的正确姿势

跨界开发者的代码：

  1. fruits = ['orange', 'grape', 'pitaya', 'blueberry']
  2. index = 0
  3. for fruit in fruits:
  4.     print(index, ':', fruit)
  5.     index += 1

Pythonic的代码：

  1. fruits = ['orange', 'grape', 'pitaya', 'blueberry']
  2. for index, fruit in enumerate(fruits):
  3.     print(index, ':', fruit)

姿势5：创建列表的正确姿势

跨界开发者的代码：

  1. data = [7, 20, 3, 15, 11]
  2. result = []
  3. for i in data:
  4.     if i > 10:
  5.         result.append(i * 3)

Pythonic的代码：

  1. data = [7, 20, 3, 15, 11]
  2. result = [num * 3 for num in data if num > 10]

姿势6：确保代码健壮性的正确姿势

跨界开发者的代码：

  1. data = {'x': '5'}
  2. if 'x' in data and isinstance(data['x'], (str, int, float)) \
  3.         and data['x'].isdigit():
  4.     value = int(data['x'])
  5.     print(value)
  6. else:
  7.     value = None

Pythonic的代码：

  1. data = {'x': '5'}
  2. try:
  3.     value = int(data['x'])
  4.     print(value)
  5. except (KeyError, TypeError, ValueError):
  6.     value = None

使用Lint工具检查你的代码规范

阅读下面的代码，看看你能看出哪些地方是有毛病的或者说不符合Python的编程规范的。

  1. from enum import *
  3. @unique
  4. class Suite (Enum):
  5.     SPADE, HEART, CLUB, DIAMOND = range(4)
  7. class Card(object):
  8.     def __init__(self,suite,face ):
  9.         self.suite = suite
  10.         self.face = face
  13.     def __repr__(self):
  14.         suites='♠♥♣♦'
  15.         faces=['','A','2','3','4','5','6','7','8','9','10','J','Q','K']
  16.         return f'{suites[self.suite.value]}{faces[self.face]}'
  18. import random
  20. class Poker(object):
  21.     def __init__(self):
  22.         self.cards =[Card(suite, face) for suite in Suite
  23.             for face in range(1, 14)]
  24.         self.current=0
  25.     def shuffle (self):
  26.         self.current=0
  27.         random.shuffle(self.cards)
  28.     def deal (self):
  29.         card = self.cards[self.current]
  30.         self.current+=1
  31.         return card
  32.     def has_next (self):
  33.         if self.current<len(self.cards): return True
  34.         return False
  36. p = Poker()
  37. p.shuffle()
  38. print(p.cards)

PyLint的安装和使用

Pylint是Python代码分析工具，它分析Python代码中的错误，查找不符合代码风格标准（默认使用的代码风格是 PEP 8）和有潜在问题的代码。

  1. pip install pylint
  2. pylint [options] module_or_package

Pylint输出格式如下所示。

模块名:行号:列号: 消息类型 消息

消息类型有以下几种：

  1. C - 惯例：违反了Python编程惯例（PEP 8）的代码。
  2. R - 重构：写得比较糟糕需要重构的代码。
  3. W - 警告：代码中存在的不影响代码运行的问题。
  4. E - 错误：代码中存在的影响代码运行的错误。
  5. F - 致命错误：导致Pylint无法继续运行的错误。

Pylint命令的常用参数：

  1. --disable=<msg ids>或-d <msg ids>：禁用指定类型的消息。
  2. --errors-only或-E：只显示错误。
  3. --rcfile=<file>：指定配置文件。
  4. --list-msgs：列出Pylint的消息清单。
  5. --generate-rcfile：生成配置文件的样例。
  6. --reports=<y_or_n>或-r <y_or_n>：是否生成检查报告。

使用Profile工具剖析你的代码性能

cProfile模块

example01.py

  1. import cProfile
  4. def is_prime(num):
  5.     for factor in range(2, int(num ** 0.5) + 1):
  6.         if num % factor == 0:
  7.             return False
  8.     return True
  11. class PrimeIter:
  13.     def __init__(self, total):
  14.         self.counter = 0
  15.         self.current = 1
  16.         self.total = total
  18.     def __iter__(self):
  19.         return self
  21.     def __next__(self):
  22.         if self.counter < self.total:
  23.             self.current += 1
  24.             while not is_prime(self.current):
  25.                 self.current += 1
  26.             self.counter += 1
  27.             return self.current
  28.         raise StopIteration()
  31. cProfile.run('list(PrimeIter(10000))')

执行结果：

  1.    114734 function calls in 0.573 seconds
  3.    Ordered by: standard name
  5.    ncalls  tottime  percall  cumtime  percall filename:lineno(function)
  6.         1    0.006    0.006    0.573    0.573 <string>:1(<module>)
  7.         1    0.000    0.000    0.000    0.000 example.py:14(__init__)
  8.         1    0.000    0.000    0.000    0.000 example.py:19(__iter__)
  9.     10001    0.086    0.000    0.567    0.000 example.py:22(__next__)
  10.    104728    0.481    0.000    0.481    0.000 example.py:5(is_prime)
  11.         1    0.000    0.000    0.573    0.573 {built-in method builtins.exec}
  12.         1    0.000    0.000    0.000    0.000 {method 'disable' of '_lsprof.Profiler' objects}

line_profiler

给需要剖析时间性能的函数加上一个profile装饰器，这个函数每行代码的执行次数和时间都会被剖析。

example02.py

  1. @profile
  2. def is_prime(num):
  3.     for factor in range(2, int(num ** 0.5) + 1):
  4.         if num % factor == 0:
  5.             return False
  6.     return True
  9. class PrimeIter:
  11.     def __init__(self, total):
  12.         self.counter = 0
  13.         self.current = 1
  14.         self.total = total
  16.     def __iter__(self):
  17.         return self
  19.     def __next__(self):
  20.         if self.counter < self.total:
  21.             self.current += 1
  22.             while not is_prime(self.current):
  23.                 self.current += 1
  24.             self.counter += 1
  25.             return self.current
  26.         raise StopIteration()
  29. list(PrimeIter(1000))

安装和使用line_profiler三方库。

  1. pip install line_profiler
  2. kernprof -lv example.py
  4. Wrote profile results to example02.py.lprof
  5. Timer unit: 1e-06 s
  7. Total time: 0.089513 s
  8. File: example02.py
  9. Function: is_prime at line 1
  11.  #      Hits         Time  Per Hit   % Time  Line Contents
  12. ==============================================================
  13.  1                                           @profile
  14.  2                                           def is_prime(num):
  15.  3     86624      43305.0      0.5     48.4      for factor in range(2, int(num ** 0.5) + 1):
  16.  4     85624      42814.0      0.5     47.8          if num % factor == 0:
  17.  5      6918       3008.0      0.4      3.4              return False
  18.  6      1000        386.0      0.4      0.4      return True

memory_profiler

给需要剖析内存性能的函数加上一个profile装饰器，这个函数每行代码的内存使用情况都会被剖析。

example03.py

  1. @profile
  2. def eat_memory():
  3.     items = []
  4.     for _ in range(1000000):
  5.         items.append(object())
  6.     return items
  9. eat_memory()

安装和使用memory_profiler三方库。

  1. pip install memory_profiler
  2. python3 -m memory_profiler example.py
  4. Filename: example03.py
  6. Line #    Mem usage    Increment   Line Contents
  7. ================================================
  8.      1   38.672 MiB   38.672 MiB   @profile
  9.      2                             def eat_memory():
  10.      3   38.672 MiB    0.000 MiB       items = []
  11.      4   68.727 MiB    0.000 MiB       for _ in range(1000000):
  12.      5   68.727 MiB    1.797 MiB           items.append(object())
  13.      6   68.727 MiB    0.000 MiB       return items

如何构建综合职业素养

学习总结

  1. 了解全局
  2. 确定范围
  3. 定义目标
  4. 寻找资源
  5. 创建学习计划
  6. 筛选资源
  7. 开始学习，浅尝辄止（YAGNI）
  8. 动手操作，边学边玩
  9. 全面掌握，学以致用
  10. 乐为人师，融会贯通

时间管理

  1. 提升专注力

  2. 充分利用碎片时间

  3. 使用番茄工作法

  4. 时间是怎么浪费掉的

  5. 任何行动都比不采取行动好

    年薪50W+的Python程序员如何写代码 - 图2

好书推荐

  1. 职业规划：《软技能 - 代码之外的生存指南》
  2. 吴军系列：《浪潮之巅》、《硅谷之谜》、《数学之美》、……
  3. 时间管理：《成为一个更高效的人》、《番茄工作法图解》