如何区分单txt文件中的两类坐标并按标识值分色绘图

Alright, let's tackle your two technical problems one by one—they're both straightforward to solve with Python, which is ideal for handling coordinate data and visualization tasks.

1. 区分单个.txt文件中的两类坐标 #

The core idea here is to filter rows based on the i value (the third column in your format) and group the corresponding (x,y) coordinates separately. Here are two common approaches:

Option 1: Using NumPy (fast for large datasets) #

NumPy makes it easy to load and filter tabular data in one go:

import numpy as np

# Load the entire text file (assumes whitespace-separated values)
data = np.loadtxt("your_data.txt")

# Extract the third column (index 2) which holds the `i` values
i_col = data[:, 2]

# Replace these with the actual two distinct `i` values from your file
target_i1 = 0
target_i2 = 1

# Create masks to filter rows for each `i` value
mask_i1 = i_col == target_i1
mask_i2 = i_col == target_i2

# Extract the (x,y) coordinates for each group (first two columns)
group_i1 = data[mask_i1][:, :2]
group_i2 = data[mask_i2][:, :2]

Option 2: Pure Python (flexible for custom formatting) #

If you need more control over parsing (e.g., handling irregular lines), use a file reader loop:

group_i1 = []
group_i2 = []

# Replace with your actual file path
with open("your_data.txt", "r") as f:
    for line in f:
        # Skip empty lines
        stripped_line = line.strip()
        if not stripped_line:
            continue
        
        # Split line into individual values (adjust separator if needed, e.g., '\t' for tabs)
        parts = stripped_line.split()
        x = float(parts[0])
        y = float(parts[1])
        i_val = int(parts[2])  # Use float if `i` is not an integer
        
        # Group coordinates based on `i` value
        if i_val == target_i1:
            group_i1.append((x, y))
        elif i_val == target_i2:
            group_i2.append((x, y))

2. 同图绘制不同i值的彩色点 #

Once you have your two coordinate groups, use matplotlib to plot them with distinct colors. Here's a complete example that ties together the data loading and plotting:

import numpy as np
import matplotlib.pyplot as plt

# Step 1: Load and filter data
data = np.loadtxt("your_data.txt")
i_col = data[:, 2]
target_i1 = 0
target_i2 = 1

mask_i1 = i_col == target_i1
mask_i2 = i_col == target_i2

x1, y1 = data[mask_i1][:, 0], data[mask_i1][:, 1]
x2, y2 = data[mask_i2][:, 0], data[mask_i2][:, 1]

# Step 2: Plot the points
plt.figure(figsize=(8, 6))
# Plot first group with red color and a label
plt.scatter(x1, y1, color="#ff4444", label=f"i = {target_i1}", s=10)
# Plot second group with blue color and a label
plt.scatter(x2, y2, color="#0099ff", label=f"i = {target_i2}", s=10)

# Add plot annotations
plt.legend(fontsize=12)
plt.xlabel("X Coordinate", fontsize=12)
plt.ylabel("Y Coordinate", fontsize=12)
plt.title("2D Points Colored by i Value", fontsize=14)

# Optional: Save the plot to a file
plt.savefig("colored_points.png", dpi=300, bbox_inches="tight")

# Show the plot
plt.show()

Key Notes: #

• Replace target_i1 and target_i2 with the actual two i values present in your file (e.g., if your i values are 5 and 10, update those variables).
• Adjust the color parameter to any valid color name/hex code you prefer.
• The s parameter controls the size of the scatter points—tweak it to make points more/less visible.

内容的提问来源于stack exchange，提问作者Ziezi