patsy equivalent of R's all.vars

Open
#155 1 comment 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Assessment

Difficulty
3/5
Estimated time
1-2 days
Newbie friendliness
35/100
Issue type
Feature
Clarity
Mostly clear
Activity status
Stale
Tech stack
python
Domain
data

Research direction

Start by tracing how patsy parses the formula strings shown in the issue. The change is done when an interface returns ['x1', 'x2', 'x3', 'x4'] for the first example and ['time'] for the cr(time, df = 3) example, including variables nested in interactions and function calls.

Written by the indexing model from the issue text.

Description

Hey patsy developers, I wonder how can we retrieve the variable names from the model formula string (in R, this can be done via the all.vars function). For example,

if I have ("y ~ x1:x2 + np.log(x3) + I(x4^2)"), I want to return a list of ['x1', 'x2', 'x3', 'x4'].
Similarly, if I have ("~cr(time, df = 3)"), I want to return a list of ['time'].

Your help is highly appreciated!

Dominant language
Python
Stars
990
Forks
106
Avg merge
7d 34m
Merged PRs (30d)
1

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

More from pydata/patsy

All issues in pydata/patsy

Similar issues

More Python issues

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.