Skip to content

CLN: annotation in reshape.merge #29490

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 9 commits into from
Nov 12, 2019
Merged

CLN: annotation in reshape.merge #29490

Changes from 1 commit
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
8384e5f
CLN: annotation in reshape.merge
jbrockmendel Nov 8, 2019
bc15a94
update per comments
jbrockmendel Nov 8, 2019
6b3f6cb
restore assignment
jbrockmendel Nov 8, 2019
a58d81e
Merge branch 'master' of https://github.com/pandas-dev/pandas into cl…
jbrockmendel Nov 10, 2019
de5fe3b
revert to generator
jbrockmendel Nov 10, 2019
731cea7
update per comment
jbrockmendel Nov 10, 2019
ed3c56c
suggested typing edit
jbrockmendel Nov 10, 2019
c33a91b
Merge branch 'master' of https://github.com/pandas-dev/pandas into cl…
jbrockmendel Nov 11, 2019
b34a9ac
update per requests
jbrockmendel Nov 11, 2019
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
99 changes: 57 additions & 42 deletions pandas/core/reshape/merge.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
import datetime
from functools import partial
import string
from typing import Optional, Tuple
import warnings

import numpy as np
Expand Down Expand Up @@ -39,6 +40,7 @@
from pandas.core.dtypes.missing import isna, na_value_for_dtype

from pandas import Categorical, Index, MultiIndex
from pandas._typing import FrameOrSeries
import pandas.core.algorithms as algos
from pandas.core.arrays.categorical import _recode_for_categories
import pandas.core.common as com
Expand All @@ -52,16 +54,16 @@
def merge(
left,
right,
how="inner",
how: str = "inner",
on=None,
left_on=None,
right_on=None,
left_index=False,
right_index=False,
sort=False,
left_index: bool = False,
right_index: bool = False,
sort: bool = False,
suffixes=("_x", "_y"),
copy=True,
indicator=False,
copy: bool = True,
indicator: bool = False,
validate=None,
):
op = _MergeOperation(
Expand All @@ -86,7 +88,9 @@ def merge(
merge.__doc__ = _merge_doc % "\nleft : DataFrame"


def _groupby_and_merge(by, on, left, right, _merge_pieces, check_duplicates=True):
def _groupby_and_merge(
by, on, left, right, _merge_pieces, check_duplicates: bool = True
):
"""
groupby & merge; we are always performing a left-by type operation

Expand Down Expand Up @@ -172,7 +176,7 @@ def merge_ordered(
right_by=None,
fill_method=None,
suffixes=("_x", "_y"),
how="outer",
how: str = "outer",
):
"""
Perform merge with optional filling/interpolation.
Expand Down Expand Up @@ -298,14 +302,14 @@ def merge_asof(
on=None,
left_on=None,
right_on=None,
left_index=False,
right_index=False,
left_index: bool = False,
right_index: bool = False,
by=None,
left_by=None,
right_by=None,
suffixes=("_x", "_y"),
tolerance=None,
allow_exact_matches=True,
allow_exact_matches: bool = True,
direction="backward",
):
"""
Expand Down Expand Up @@ -541,23 +545,23 @@ class _MergeOperation:

def __init__(
self,
left,
right,
how="inner",
left: FrameOrSeries,
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Haven't debugged to verify but we can mix and match objects here right? So something as follows:

>>> df = pd.DataFrame([[1]])
>>> ser = pd.Series([1], name="a")
>>> pd.merge(df, ser, left_index=True, right_index=True)
   0  a
0  1  1

If that's the case then these actually should be Union[DataFrame, Series] instead of FrameOrSeries. Pretty tricky

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Correct, left and right can each be either DataFrame or Series, i.e. 4 valid combinations. Does FrameOrSeries assume you're not mix-and-matching? will update

right: FrameOrSeries,
how: str = "inner",
on=None,
left_on=None,
right_on=None,
axis=1,
left_index=False,
right_index=False,
sort=True,
left_index: bool = False,
right_index: bool = False,
sort: bool = True,
suffixes=("_x", "_y"),
copy=True,
indicator=False,
copy: bool = True,
indicator: bool = False,
validate=None,
):
left = validate_operand(left)
right = validate_operand(right)
left = _validate_operand(left)
right = _validate_operand(right)
self.left = self.orig_left = left
self.right = self.orig_right = right
self.how = how
Expand All @@ -577,7 +581,7 @@ def __init__(
self.indicator = indicator

if isinstance(self.indicator, str):
self.indicator_name = self.indicator
self.indicator_name = self.indicator # type: Optional[str]
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

use py3.6 syntax :)

my preference would be to declare outside the if else.

elif isinstance(self.indicator, bool):
self.indicator_name = "_merge" if self.indicator else None
else:
Expand Down Expand Up @@ -658,7 +662,9 @@ def get_result(self):

return result

def _indicator_pre_merge(self, left, right):
def _indicator_pre_merge(
self, left: ABCDataFrame, right: ABCDataFrame
) -> Tuple[ABCDataFrame, ABCDataFrame]:

columns = left.columns.union(right.columns)

Expand Down Expand Up @@ -878,7 +884,12 @@ def _get_join_info(self):
return join_index, left_indexer, right_indexer

def _create_join_index(
self, index, other_index, indexer, other_indexer, how="left"
self,
index: Index,
other_index: Index,
indexer,
other_indexer,
how: str = "left",
):
"""
Create a join index by rearranging one index to match another
Expand Down Expand Up @@ -1263,7 +1274,9 @@ def _validate(self, validate: str):
raise ValueError("Not a valid argument for validate")


def _get_join_indexers(left_keys, right_keys, sort=False, how="inner", **kwargs):
def _get_join_indexers(
left_keys, right_keys, sort: bool = False, how: str = "inner", **kwargs
):
"""

Parameters
Expand Down Expand Up @@ -1410,13 +1423,13 @@ def __init__(
on=None,
left_on=None,
right_on=None,
left_index=False,
right_index=False,
left_index: bool = False,
right_index: bool = False,
axis=1,
suffixes=("_x", "_y"),
copy=True,
copy: bool = True,
fill_method=None,
how="outer",
how: str = "outer",
):

self.fill_method = fill_method
Expand Down Expand Up @@ -1508,18 +1521,18 @@ def __init__(
on=None,
left_on=None,
right_on=None,
left_index=False,
right_index=False,
left_index: bool = False,
right_index: bool = False,
by=None,
left_by=None,
right_by=None,
axis=1,
suffixes=("_x", "_y"),
copy=True,
copy: bool = True,
fill_method=None,
how="asof",
how: str = "asof",
tolerance=None,
allow_exact_matches=True,
allow_exact_matches: bool = True,
direction="backward",
):

Expand Down Expand Up @@ -1757,13 +1770,15 @@ def flip(xs):
return func(left_values, right_values, self.allow_exact_matches, tolerance)


def _get_multiindex_indexer(join_keys, index, sort):
def _get_multiindex_indexer(join_keys, index: MultiIndex, sort: bool):

# bind `sort` argument
fkeys = partial(_factorize_keys, sort=sort)

# left & right join labels and num. of levels at each location
rcodes, lcodes, shape = map(list, zip(*map(fkeys, index.levels, join_keys)))
mapped = [fkeys(index.levels[n], join_keys[n]) for n in range(len(index.levels))]
zipped = zip(*mapped)
rcodes, lcodes, shape = [list(x) for x in zipped]
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

was this cleaning or silencing mypy?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

this was necessary to silence mypy. i also dont like the pattern that was used before (and am not wild about the zip still here, open to suggestions)

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

yep. fine with this. just curious.

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

OK with splitting this out but should probably keep mapped as a generator; coercing to list may have some overhead

if sort:
rcodes = list(map(np.take, rcodes, index.codes))
else:
Expand Down Expand Up @@ -1791,7 +1806,7 @@ def _get_multiindex_indexer(join_keys, index, sort):
return libjoin.left_outer_join(lkey, rkey, count, sort=sort)


def _get_single_indexer(join_key, index, sort=False):
def _get_single_indexer(join_key, index, sort: bool = False):
left_key, right_key, count = _factorize_keys(join_key, index, sort=sort)

left_indexer, right_indexer = libjoin.left_outer_join(
Expand All @@ -1801,7 +1816,7 @@ def _get_single_indexer(join_key, index, sort=False):
return left_indexer, right_indexer


def _left_join_on_index(left_ax, right_ax, join_keys, sort=False):
def _left_join_on_index(left_ax: Index, right_ax: Index, join_keys, sort: bool = False):
if len(join_keys) > 1:
if not (
(isinstance(right_ax, MultiIndex) and len(join_keys) == right_ax.nlevels)
Expand Down Expand Up @@ -1915,7 +1930,7 @@ def _factorize_keys(lk, rk, sort=True):
return llab, rlab, count


def _sort_labels(uniques, left, right):
def _sort_labels(uniques: np.ndarray, left, right):
if not isinstance(uniques, np.ndarray):
# tuplesafe
uniques = Index(uniques).values
Expand All @@ -1930,7 +1945,7 @@ def _sort_labels(uniques, left, right):
return new_left, new_right


def _get_join_keys(llab, rlab, shape, sort):
def _get_join_keys(llab, rlab, shape, sort: bool):

# how many levels can be done without overflow
pred = lambda i: not is_int64_overflow_possible(shape[:i])
Expand Down Expand Up @@ -1970,7 +1985,7 @@ def _any(x) -> bool:
return x is not None and com.any_not_none(*x)


def validate_operand(obj):
def _validate_operand(obj: FrameOrSeries) -> ABCDataFrame:
if isinstance(obj, ABCDataFrame):
return obj
elif isinstance(obj, ABCSeries):
Expand All @@ -1985,7 +2000,7 @@ def validate_operand(obj):
)


def _items_overlap_with_suffix(left, lsuffix, right, rsuffix):
def _items_overlap_with_suffix(left: Index, lsuffix, right: Index, rsuffix):
"""
If two indices overlap, add suffixes to overlapping entries.

Expand Down