Reformatting

Author: Reksbril

pandas/core/frame.py

@@ -6564,7 +6564,7 @@ def apply(self, func, axis=0, raw=False, result_type=None, args=(), **kwds):
         """
         from pandas.core.apply import frame_apply
 
-        #Old apply function, which will be used for each part of DataFrame
+        # Old apply function, which will be used for each part of DataFrame
         def partial_apply(dataframe):
             op = frame_apply(
                 dataframe,
@@ -6580,7 +6580,7 @@ def partial_apply(dataframe):
         def get_dtype(dataframe, column):
             return dataframe.dtypes.values[column]
 
-        if axis == 0 or axis == 'index':
+        if axis == 0 or axis == "index":
             if self.shape[1] == 0:
                 return partial_apply(self)
 
@@ -6596,10 +6596,13 @@ def get_dtype(dataframe, column):
                 type = get_dtype(self, i)
                 j = i + 1
 
-                #While the dtype of column is the same as previous ones, they are handled together
-                while j < self.shape[1] and pandas.core.dtypes.common.is_dtype_equal(type, get_dtype(self, j)):
+                # While the dtype of column is the same as previous ones,
+                # they are handled together
+                while j < self.shape[1] and pandas.core.dtypes.common.is_dtype_equal(
+                    type, get_dtype(self, j)
+                ):
                     j += 1
-                frame = self.iloc[:, i: j]
+                frame = self.iloc[:, i:j]
                 i = j
                 result = partial_apply(frame)
 
@@ -6619,7 +6622,6 @@ def get_dtype(dataframe, column):
         else:
             return partial_apply(self)
 
-
     def applymap(self, func):
         """
         Apply a function to a Dataframe elementwise.

pandas/tests/frame/test_apply.py

@@ -693,11 +693,8 @@ def test_apply_dup_names_multi_agg(self):
 
     def test_apply_get_dtype(self):
         # GH 28773
-        df = DataFrame({
-            "col_1": [1, 2, 3],
-            "col_2": ["hi", "there", "friend"]
-        })
-        expected = Series(data=['int64', 'object'] ,index=['col_1', 'col_2'])
+        df = DataFrame({"col_1": [1, 2, 3], "col_2": ["hi", "there", "friend"]})
+        expected = Series(data=["int64", "object"], index=["col_1", "col_2"])
         tm.assert_series_equal(df.apply(lambda x: x.dtype), expected)