如何用python绘制按文件类型划分的折线图？

import matplotlib.pyplot as plt
import os

def count_file_types(directory):
    file_types = {}
    for root, dirs, files in os.walk(directory):
        for file in files:
            file_name, file_extension = os.path.splitext(file)
            if file_extension not in file_types:
                file_types[file_extension] = 1
            else:
                file_types[file_extension] += 1
    return file_types

directory = '/path/to/directory'  # 替换为实际的目录路径
file_types = count_file_types(directory)

types = list(file_types.keys())
counts = list(file_types.values())

plt.plot(types, counts)
plt.xlabel('File Types')
plt.ylabel('Count')
plt.title('File Types Distribution')
plt.xticks(rotation=90)
plt.show()