新列 - 来自多个行和列的多个条件

发布于 2025-01-17 09:23:09 字数 1412 浏览 1 评论 0原文

在带有三列的大数据框中，我必须根据前三列添加一个带有条件值的第四列。

这是关于旅行的。一个人（ID）有几种旅行选择（例如巴士，汽车，火车，...）。有时，一个人可以通过火车从A到B旅行几种选择。

我的数据框看起来像这样：

main_mode	持续时间	ID
火车	1250	1
火车	900	1
车	540	1
火车	650	2
汽车	450	2
平面	350	2
火车	350	3
CAR	890	3

现在我想创建一个新的列“ Train_avail”。如果火车旅行的时间是汽车之旅的两倍，那么我想在新列中写下train_avail = 0，否则写1。

我想拥有的东西：

main_mode	distration	id	train_avail
train	1250	1	0
火车	900	1 1 1	1 1
辆车	540	1	NA
火车	650	2	1
车	450	2	NA
平面	350	2	NA
火车	350	3	1
CAR	890	3	NA

感谢您的帮助！

原文

In a large dataframe with three columns I have to add a fourth column with values conditionally based on the first three columns.

It's about travel. A person (id) has several options to travel (e.g. bus, car, train, ...). Sometimes a person has several options by train to travel from A to B for example.

My dataframe looks like this:

main_mode	duration	id
train	1250	1
train	900	1
car	540	1
train	650	2
car	450	2
plane	350	2
train	350	3
car	890	3

Now I want to create a new column "train_avail". If the train journey takes twice as long as the car journey, then I would like to write train_avail = 0 in a new column otherwise write 1.

What i would like to have:

main_mode	duration	id	train_avail
train	1250	1	0
train	900	1	1
car	540	1	NA
train	650	2	1
car	450	2	NA
plane	350	2	NA
train	350	3	1
car	890	3	NA

Thanks for your help!!

分享到QQ

分享到微博

如果你对这篇内容有疑问，欢迎到本站社区发帖提问参与讨论，获取更多帮助，或者扫码二维码加入 Web 技术交流群。

发布评论

需要登录才能够评论，你可以免费注册一个本站的账号。

难理解 2025-01-24 09:23:09

df |>
group_by(id) |>
  mutate(train_avail = {
    car <- min(duration[main_mode == "car"])
    if_else(main_mode == "train",
            if_else(duration > 2* car, 0, 1),
            NA_real_)
    })

##> + # A tibble: 8 × 4
##> # Groups:   id [3]
##>   main_mode duration    id train_avail
##>   <chr>        <int> <int>       <dbl>
##> 1 train         1250     1           0
##> 2 train          900     1           1
##> 3 car            540     1          NA
##> 4 train          650     2           1
##> 5 car            450     2          NA
##> 6 plane          350     2          NA
##> 7 train          350     3           1
##> 8 car            890     3          NA

df |>
group_by(id) |>
  mutate(train_avail = {
    car <- min(duration[main_mode == "car"])
    if_else(main_mode == "train",
            if_else(duration > 2* car, 0, 1),
            NA_real_)
    })

##> + # A tibble: 8 × 4
##> # Groups:   id [3]
##>   main_mode duration    id train_avail
##>   <chr>        <int> <int>       <dbl>
##> 1 train         1250     1           0
##> 2 train          900     1           1
##> 3 car            540     1          NA
##> 4 train          650     2           1
##> 5 car            450     2          NA
##> 6 plane          350     2          NA
##> 7 train          350     3           1
##> 8 car            890     3          NA

回复收藏 0 原文

~没有更多了~