旋转坐标系下数据点轴线的定位与绘制方法咨询

嗨，我来帮你搞定这个问题！你遇到的核心问题是：线性回归是用来拟合变量间依赖关系的，但这里你的数据只是做了旋转，变量间并没有依赖关系，所以线性回归没法找到数据分布的主轴。这时候我们需要用主成分分析（PCA），它能精准捕捉数据方差最大的方向——也就是你要找的那条轴线。

核心思路 #

旋转后的点集，其分布的主轴就是原始数据的x轴经过旋转后的方向。PCA可以自动锁定这个方差最大的方向，同时我们可以通过数据的均值来确定旋转中心（因为你是围绕c(0.5, 0)旋转的，均值点会非常接近这个中心）。

具体实现步骤 #

下面结合你的代码给出完整解决方案：

1. 加载包并生成/旋转数据（保留你原来的代码）

library(ggplot2)
library(rearrr)
library(stats)  # 用于PCA分析

# 生成原始均匀分布数据
x <- runif(2000, min = 0, max = 1)
y <- runif(2000, min = -0.4, max = 0.4)
df <- data.frame(x = x, y = y)

# 围绕中心点旋转数据
dfr <- rotate_2d(df, -30, origin = c(0.5, 0), x_col = "x", y_col = "y")

2. 计算中心点与PCA分析

# 计算旋转后数据的中心点（和原始旋转原点高度接近）
center <- colMeans(dfr[, c("x_rotated", "y_rotated")])

# 对旋转后的数据执行PCA，锁定方差最大的方向
pca <- prcomp(dfr[, c("x_rotated", "y_rotated")], center = TRUE, scale. = FALSE)

# 提取第一主成分的方向向量（就是我们要找的轴线方向）
pc1 <- pca$rotation[, 1]

3. 计算旋转角度并绘制轴线
   我们可以从主成分的方向向量算出轴线与x轴的夹角，再绘制穿过中心点的这条红色轴线：

# 将方向向量转换为角度（弧度转角度）
angle <- atan2(pc1[2], pc1[1]) * (180/pi)

# 生成轴线上的两个端点（从中心点向两侧延伸，长度可调整）
axis_points <- data.frame(
  x = center[1] + c(-1, 1) * pc1[1] * 0.8,
  y = center[2] + c(-1, 1) * pc1[2] * 0.8
)

# 绘图：散点+线性回归（对比用）+红色主轴
ggplot(dfr) +
  geom_point(aes(x_rotated, y_rotated), alpha = 0.5) +
  geom_smooth(aes(x_rotated, y_rotated), method = "lm", formula = y ~ x, color = "blue", linetype = "dashed") +
  geom_line(data = axis_points, aes(x, y), color = "red", size = 1) +
  labs(title = "旋转后数据的主轴（红色）与线性回归对比（蓝色虚线）") +
  theme_minimal()

为什么这个方法有效？ #

• PCA的第一主成分就是数据分布最“舒展”的方向，正好对应原始x轴旋转后的方向，也就是你要找的目标轴线。
• 用数据均值计算中心点，是因为旋转操作围绕固定点进行，旋转后数据的均值会非常接近原始旋转原点（无其他变换时完全重合）。

这样你就能准确找到旋转角度和中心点，画出那条红色的轴线啦！

备注：内容来源于stack exchange，提问作者Tianjian Qin