6 examples of 'pandas add new column based on another column' in Python

160
def __add__(self, other):
161
    if isinstance(self.spark_type, StringType):
162
        # Concatenate string columns
163
        if isinstance(other, IndexOpsMixin) and isinstance(other.spark_type, StringType):
164
            return _column_op(F.concat)(self, other)
165
        # Handle df['col'] + 'literal'
166
        elif isinstance(other, str):
167
            return _column_op(F.concat)(self, F.lit(other))
168
        else:
169
            raise TypeError('string addition can only be applied to string series or literals.')
170
    else:
171
        return _column_op(spark.Column.__add__)(self, other)

813
def merge(old_cols, new_cols):
814
    return old_cols + new_cols

136
def copy_column(self, to, fro):
137
    """Copy column data
138
139
    Args:
140
        fro (str): column name to copy data from
141
        to (str): column name to copy to
142
    """
143
    logging.debug("copying {} to {}".format(fro, to))
144
    self.df[to] = self.df[fro]

463
def __add__(self, other, option='left'):
464
    if isinstance(other, Column):
465
        b = [other]
466
    elif isinstance(other, ColDefs):
467
        b = list(other.columns)
468
    else:
469
        raise TypeError('Wrong type of input.')
470
    if option == 'left':
471
        tmp = list(self.columns) + b
472
    else:
473
        tmp = b + list(self.columns)
474
    return ColDefs(tmp)

181
def _create_column(self, name):
182
    column = Column(name, self._ids, storagegroup=self.storagegroup)
183
    index = len(self._columns)
184
    self._columndict[name] = index
185
    self._columns.append(column)
186
    return column

38
def __add__(self, other):
39
40
    for attr, data in self:
41
        data_other = getattr(other, attr)
42
43
        merged = pandas.concat(
44
            [data, data_other], sort=False
45
        )
46
47
        setattr(self, attr, merged)
48
49
    self.update_iid()
50
51
    return self