python 读取文件两列画散点图-CFANZ编程社区

Python读取文件两列画散点图

在数据分析和可视化中，散点图是一种常用的图表类型，用于展示两个变量之间的关系。Python作为一种强大的编程语言，拥有许多库可以用来读取文件并绘制散点图。本文将介绍如何使用Python读取文件中的两列数据，并使用matplotlib库绘制散点图。

读取文件

在Python中，我们可以使用open()函数来打开一个文件，并使用readlines()方法读取文件的每一行。下面是一个示例代码，用于读取一个名为data.txt文件中的数据。

with open('data.txt', 'r') as file:
    lines = file.readlines()

在读取文件的过程中，我们可以通过逐行遍历数据来处理每一行的内容。

数据处理

在读取文件后，我们需要对数据进行处理，以便用于绘制散点图。一般来说，文件中的数据可能包含标题行、缺失值或其他非数值字符。我们需要首先去除标题行，并将数据转换为数值类型。

x_values = []
y_values = []

for line in lines[1:]:
    line = line.strip()
    if line:
        x, y = line.split()
        x_values.append(float(x))
        y_values.append(float(y))

上面的代码中，我们分别定义了两个空列表x_values和y_values，用于保存从文件中读取的数据。在遍历每一行数据时，我们首先使用strip()方法去除行两端的空白字符。然后，我们使用split()方法将一行数据拆分成两个部分，分别赋给变量x和y。最后，我们使用float()函数将字符串转换为浮点数，并将转换后的值添加到相应的列表中。

绘制散点图

使用matplotlib库可以方便地绘制散点图。首先，我们需要导入matplotlib.pyplot模块，并使用scatter()函数绘制散点图。

import matplotlib.pyplot as plt

plt.scatter(x_values, y_values)
plt.xlabel('X')
plt.ylabel('Y')
plt.title('Scatter Plot')
plt.show()

在上面的代码中，我们首先使用plt.scatter()函数绘制散点图，传入两个列表x_values和y_values作为参数。然后，我们使用plt.xlabel()和plt.ylabel()函数分别设置x轴和y轴的标签。最后，我们使用plt.title()函数设置图表的标题。最后一行代码plt.show()用于显示散点图。

完整代码示例

下面是一个完整的示例代码，用于读取文件中的两列数据，并绘制散点图。

import matplotlib.pyplot as plt

with open('data.txt', 'r') as file:
    lines = file.readlines()

x_values = []
y_values = []

for line in lines[1:]:
    line = line.strip()
    if line:
        x, y = line.split()
        x_values.append(float(x))
        y_values.append(float(y))

plt.scatter(x_values, y_values)
plt.xlabel('X')
plt.ylabel('Y')
plt.title('Scatter Plot')
plt.show()