\documentclass[a4paper]{book}
\usepackage{ctex}
\usepackage[dvipsnames]{xcolor}
\usepackage{listings}

\lstset{
    basicstyle=\ttfamily, % 设置字体族
    breaklines=true, % 自动换行
    keywordstyle=\bfseries\color{NavyBlue}, % 设置关键字为粗体，颜色为 NavyBlue
    emph=[1]{}, % 指定强调词，如果有多个，用逗号隔开
    %emph=[2]{as},
    emphstyle=[1]\bfseries\color{Rhodamine}, % 强调词样式设置
    %emphstyle=[2]\bfseries\color{NavyBlue},
    commentstyle=\itshape\color{black!50!white}, % 设置注释样式，斜体，浅灰色
    stringstyle=\bfseries\color{PineGreen!90!black}, % 设置字符串样式
    columns=flexible,
    numbers=left, % 显示行号在左边
    numbersep=1em, % 设置行号的具体位置
    numberstyle=\footnotesize, % 缩小行号
    frame=single, % 边框
    framesep=0em % 设置代码与边框的距离
}

\begin{document}

\begin{lstlisting}[language=Python]
    import matplotlib.pyplot as plt # 绘制图形的库
    from matplotlib_venn import venn2 # 绘制二元韦恩图的库，如果是三元则是 venn3

    plt.figure(figsize=(8, 8)) # 初始化设置图形的尺寸
\end{lstlisting}

\end{document}