Skip to content

Commit 5e942fd

Browse files
committed
Refactor multi.py to handle single tickers and rename columns
1 parent 16072e8 commit 5e942fd

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

yfinance/multi.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -34,10 +34,11 @@
3434

3535

3636
@utils.log_indent_decorator
37-
def download(tickers, start=None, end=None, actions=False, threads=True, ignore_tz=None,
38-
group_by='column', auto_adjust=False, back_adjust=False, repair=False, keepna=False,
39-
progress=True, period="max", interval="1d", prepost=False,
40-
proxy=None, rounding=False, timeout=10, session=None, return_multi_index=True):
37+
def download(tickers, start=None, end=None, actions=False, threads=True,
38+
ignore_tz=None, group_by='column', auto_adjust=False, back_adjust=False,
39+
repair=False, keepna=False, progress=True, period="max", interval="1d",
40+
prepost=False, proxy=None, rounding=False, timeout=10, session=None,
41+
multi_level_index=True):
4142
"""Download yahoo tickers
4243
:Parameters:
4344
tickers : str, list
@@ -85,7 +86,7 @@ def download(tickers, start=None, end=None, actions=False, threads=True, ignore_
8586
seconds. (Can also be a fraction of a second e.g. 0.01)
8687
session: None or Session
8788
Optional. Pass your own session object to be used for all requests
88-
return_multi_index: bool
89+
multi_level_index: bool
8990
Optional. Always return a MultiIndex DataFrame? Default is False
9091
"""
9192
logger = utils.get_yf_logger()
@@ -217,7 +218,7 @@ def download(tickers, start=None, end=None, actions=False, threads=True, ignore_
217218
data.columns = data.columns.swaplevel(0, 1)
218219
data.sort_index(level=0, axis=1, inplace=True)
219220

220-
if not return_multi_index and len(tickers) == 1:
221+
if not multi_level_index and len(tickers) == 1:
221222
data = data.droplevel(0 if group_by == 'ticker' else 1, axis=1).rename_axis(None, axis=1)
222223

223224
return data

0 commit comments

Comments
 (0)