Skip to content

Commit 362db19

Browse files
committed
PERF: perf fixes
1 parent 744daa7 commit 362db19

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

pandas/core/common.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -952,7 +952,8 @@ def diff(arr, n, axis=0):
952952
out_arr[res_indexer] = arr[res_indexer] - arr[lag_indexer]
953953

954954
if is_timedelta:
955-
out_arr = lib.map_infer(out_arr.ravel(),lib.Timedelta).reshape(out_arr.shape)
955+
from pandas import TimedeltaIndex
956+
out_arr = TimedeltaIndex(out_arr.ravel().astype('int64')).asi8.reshape(out_arr.shape).astype('timedelta64[ns]')
956957

957958
return out_arr
958959

pandas/core/groupby.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -2993,8 +2993,9 @@ def transform(self, func, *args, **kwargs):
29932993
type(result.index) != type(obj.index)) or
29942994
len(result.index) != len(obj.index)):
29952995
results = obj.values.copy()
2996+
indices = self.indices
29962997
for (name, group), (i, row) in zip(self, result.iterrows()):
2997-
indexer = self._get_index(name)
2998+
indexer = indices[name]
29982999
results[indexer] = np.tile(row.values,len(indexer)).reshape(len(indexer),-1)
29993000
return DataFrame(results,columns=result.columns,index=obj.index).convert_objects()
30003001

0 commit comments

Comments
 (0)