如何在 R 中重现以下涉及 for 循环的 matlab 代码？

Question

我想在 R 中使用 for 循环绘制图形。我尝试使用 matlab，它可以正常工作。

我在matlab中尝试过以下代码：

# matlab code

d=0:400;
for h=1:10:50
w=1./(1+(d.^2/h^2));
plot(d,w);
hold on
end
xlabel('Distance(km)')
ylabel('Weight')

我在 R 中尝试了以下代码，但它似乎不起作用

# R code

h <- c(1,20,50,100,200,400)
d <- seq(40, 420, by = 20)
for (i in 1:d) {
  w <- 1/(1+(i^2/h^2))
  lines(d, w)
}

如何使用 for 循环在 R 中重现上述 matlab 代码？

Answer 1

这是使用 Tidyverse 包的一种可能性：

library(tidyverse)

df <- 
  tibble(
    d = rep(0:400, 5),
    h = rep(c(1, 11, 21, 31, 41), each = 401)
  ) %>% 
  mutate(w = 1 / (1 + (d ^ 2 / h ^ 2)))

df %>% 
  mutate(h = factor(h)) %>% 
  ggplot(aes(x = d, y = w, color = h)) +
  geom_line() + 
  labs(
    x = "Distance (km)",
    y = "Weight"
  ) +
  theme(legend.position = "bottom")