Both methods you mentioned are useful, but combining them can make rhythm imitation easier. First, listen and feel the rhythm naturally (internalize the groove), then lightly count or subdivide beats only where tricky passages occur. Practicing with a metronome and clapping or tapping before playing...